Searched refs:CopyCPUTensorToDeviceSync (Results 1 – 11 of 11) 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 | 85 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/compiler/jit/ |
| D | xla_device.cc | 492 device_context->CopyCPUTensorToDeviceSync(&parsed, this, ©)); in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | permuter_test.cc | 364 TF_CHECK_OK(dev_info->default_context->CopyCPUTensorToDeviceSync( in InitTensor()
|
| D | ring_gatherer_test.cc | 441 ->CopyCPUTensorToDeviceSync(&cpu_tensor, device_, in InitTensor()
|
| D | ring_reducer_test.cc | 467 ->CopyCPUTensorToDeviceSync(&cpu_tensor, device_, &tensor_) in InitTensor()
|
| D | hierarchical_tree_broadcaster_test.cc | 622 TF_CHECK_OK(dev_info->default_context->CopyCPUTensorToDeviceSync( in InitTensor()
|
| D | process_function_library_runtime_test.cc | 202 ->CopyCPUTensorToDeviceSync(&cpu_tensor, gpu_device_, in CPUToGPU()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_test.cc | 106 TF_ASSERT_OK(device_context->CopyCPUTensorToDeviceSync(cpu_tensor, device, in CopyCPUToGPU()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 280 TF_CHECK_OK(dev_info->default_context->CopyCPUTensorToDeviceSync( in InitTensor()
|