Home
last modified time | relevance | path

Searched defs:device_tensor (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device_context.cc27 Tensor *device_tensor, in CopyCPUTensorToDevice()
102 void SYCLDeviceContext::CopyDeviceTensorToCPU(const Tensor *device_tensor, in CopyDeviceTensorToCPU()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util_platform_specific.cc28 Tensor* device_tensor, in CopyCPUTensorToDevice()
33 void GPUDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc108 Tensor* device_tensor, in CopyCPUTensorToDevice()
199 void XlaDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()
Dxla_compile_on_demand_op.cc119 const Tensor& device_tensor = ctx->input(i); in Compile() local
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h80 Tensor* device_tensor, in CopyCPUTensorToDevice()
95 virtual void CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_testlib.cc33 Tensor* device_tensor = nullptr; in Compute() local
Dprocess_function_library_runtime_test.cc105 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/
Dtrt_engine_op.cc316 const auto device_tensor = in ExecuteCalibration() local
669 Tensor* device_tensor; in AllocateCalibrationResources() local
/external/tensorflow/tensorflow/core/kernels/
Dstack.cc305 Tensor* device_tensor = in ComputeAsync() local