Searched refs:CopyDeviceTensorToCPUSync (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.cc | 37 Status DeviceContext::CopyDeviceTensorToCPUSync(const Tensor* device_tensor, in CopyDeviceTensorToCPUSync() function in tensorflow::DeviceContext
|
| D | device_base.h | 105 Status CopyDeviceTensorToCPUSync(const Tensor* device_tensor,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | unary_ops_composition_test.cc | 102 TF_CHECK_OK(device_context->CopyDeviceTensorToCPUSync( in RunComposedOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster_test.cc | 453 TF_CHECK_OK(dev_info->default_context->CopyDeviceTensorToCPUSync( in RunTest()
482 TF_CHECK_OK(dev_info->default_context->CopyDeviceTensorToCPUSync( in RunTest()
|
| D | ring_alg.cc | 259 Status st = gpu_device_info->default_context->CopyDeviceTensorToCPUSync( in TensorDebugString()
|
| D | permuter_test.cc | 289 TF_CHECK_OK(dev_info->default_context->CopyDeviceTensorToCPUSync( in RunTest()
|
| D | ring_gatherer_test.cc | 322 ->CopyDeviceTensorToCPUSync(inst, "" /*tensor_name*/, dev, in RunTest()
|
| D | ring_reducer_test.cc | 344 ->CopyDeviceTensorToCPUSync(inst, "" /*tensor_name*/, dev, in RunTest()
|
| D | process_function_library_runtime_test.cc | 183 ->CopyDeviceTensorToCPUSync(&device_tensor, "", gpu_device_, in GPUToCPU()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_test.cc | 112 TF_ASSERT_OK(device_context->CopyDeviceTensorToCPUSync( in CopyGPUToCPU()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 217 TF_CHECK_OK(dev_info->default_context->CopyDeviceTensorToCPUSync( in RunTest()
|
| D | functional_ops.cc | 415 TF_RETURN_IF_ERROR(device_ctx->CopyDeviceTensorToCPUSync( in CondResultToBool()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 643 TF_RETURN_IF_ERROR(device_context->CopyDeviceTensorToCPUSync( in BuildXlaCompilerArguments()
|