Searched defs:device_tensor (Results 1 – 14 of 14) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_util_platform_specific.cc | 28 Tensor* device_tensor, in CopyCPUTensorToDevice()
35 void GPUDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_context.cc | 28 const Tensor* cpu_tensor, Device* device, Tensor* device_tensor, in CopyCPUTensorToDevice()
34 void PluggableDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()
|
| H A D | pluggable_device_util.cc | 176 const Tensor* device_tensor, Tensor* cpu_tensor, StatusCallback done) { in CopyPluggableDeviceTensorToCPU()
223 Device* device, Tensor* device_tensor, StatusCallback done, in CopyCPUTensorToPluggableDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | virtual_device.cc | 40 Tensor* device_tensor, in CopyCPUTensorToDevice()
47 void VirtualDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_device_context.cc | 116 Tensor* device_tensor, in CopyCPUTensorToDevice()
209 void XlaDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | serialization.cpp | 241 auto device_tensor = at::from_blob( in THPStorage_writeFileRaw() local
390 auto device_tensor = at::from_blob( in THPStorage_readFileRaw() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | device_base.cc | 37 Status DeviceContext::CopyDeviceTensorToCPUSync(const Tensor* device_tensor, in CopyDeviceTensorToCPUSync()
|
| H A D | device_base.h | 99 virtual void CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | function_testlib.cc | 33 Tensor* device_tensor = nullptr; in Compute() local
|
| H A D | process_function_library_runtime_test.cc | 176 Tensor GPUToCPU(const Tensor& device_tensor) { in GPUToCPU()
201 Tensor device_tensor(gpu_device_->GetAllocator({}), cpu_tensor.dtype(), in CPUToGPU() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | ts_backend_impl.cpp | 102 auto device_tensor = at::full(tensor.sizes(), tensor.item(), options); in MakeComputationDataFromTensor() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | stack.cc | 306 Tensor* device_tensor = in ComputeAsync() local
|
| H A D | resource_variable_ops.cc | 941 Status CopyTensorToHost(OpKernelContext* c, const Tensor& device_tensor, in CopyTensorToHost()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_op.cc | 671 const auto device_tensor = &calib_ctx->device_tensors_.at(i); in ExecuteCalibration() local
|