Searched defs:device_tensor (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/sycl/ |
D | sycl_device_context.cc | 27 Tensor *device_tensor, in CopyCPUTensorToDevice() 102 void SYCLDeviceContext::CopyDeviceTensorToCPU(const Tensor *device_tensor, in CopyDeviceTensorToCPU()
|
/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
D | gpu_util_platform_specific.cc | 28 Tensor* device_tensor, in CopyCPUTensorToDevice() 33 void GPUDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | xla_device_context.cc | 108 Tensor* device_tensor, in CopyCPUTensorToDevice() 199 void XlaDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()
|
D | xla_compile_on_demand_op.cc | 119 const Tensor& device_tensor = ctx->input(i); in Compile() local
|
/external/tensorflow/tensorflow/core/framework/ |
D | device_base.h | 80 Tensor* device_tensor, in CopyCPUTensorToDevice() 95 virtual void CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | function_testlib.cc | 33 Tensor* device_tensor = nullptr; in Compute() local
|
D | process_function_library_runtime_test.cc | 105 Tensor GPUToCPU(const Tensor& device_tensor) { in GPUToCPU() 138 Tensor device_tensor(gpu_device_->GetAllocator({}), cpu_tensor.dtype(), in CPUToGPU() local
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
D | trt_engine_op.cc | 316 const auto device_tensor = in ExecuteCalibration() local 669 Tensor* device_tensor; in AllocateCalibrationResources() local
|
/external/tensorflow/tensorflow/core/kernels/ |
D | stack.cc | 305 Tensor* device_tensor = in ComputeAsync() local
|