Searched refs:CopyCPUTensorToDeviceSync (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.cc | 52 Status DeviceContext::CopyCPUTensorToDeviceSync(const Tensor* cpu_tensor, in CopyCPUTensorToDeviceSync() function in tensorflow::DeviceContext
|
| D | device_base.h | 86 Status CopyCPUTensorToDeviceSync(const Tensor* cpu_tensor, Device* device,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | unary_ops_composition_test.cc | 91 TF_CHECK_OK(device_context->CopyCPUTensorToDeviceSync(&input_on_host, in RunComposedOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_test_util.cc | 270 TF_CHECK_OK(dev_info->default_context->CopyCPUTensorToDeviceSync( in CopyTensorToDevice()
|
| D | process_function_library_runtime_test.cc | 204 ->CopyCPUTensorToDeviceSync(&cpu_tensor, gpu_device_, in CPUToGPU()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 525 device_context->CopyCPUTensorToDeviceSync(&parsed, this, ©)); in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 282 TF_CHECK_OK(dev_info->default_context->CopyCPUTensorToDeviceSync( in InitTensor()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_test.cc | 116 TF_ASSERT_OK(device_context->CopyCPUTensorToDeviceSync(cpu_tensor, device, in CopyCPUToGPU()
|