Searched refs:gpu_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 257 const Tensor* gpu_tensor, Tensor* cpu_tensor, in CopyGPUTensorToCPU() argument
262 Status s = PrepareCopy(gpu_device, device_context, *gpu_tensor, cpu_tensor, in CopyGPUTensorToCPU()
279 const int64 total_bytes = gpu_tensor->TotalBytes(); in CopyGPUTensorToCPU()
281 void* src_ptr = GetBase(gpu_tensor); in CopyGPUTensorToCPU()
287 TensorReference input_ref(*gpu_tensor); in CopyGPUTensorToCPU()
302 Device* gpu_device, Tensor* gpu_tensor, in CopyCPUTensorToGPU() argument
307 Status s = PrepareCopy(gpu_device, device_context, *cpu_tensor, gpu_tensor, in CopyCPUTensorToGPU()
328 void* dst_ptr = GetBase(gpu_tensor); in CopyCPUTensorToGPU()
|
| D | gpu_util.h | 49 const Tensor* gpu_tensor, Tensor* cpu_tensor,
90 Device* gpu_device, Tensor* gpu_tensor,
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 64 gpu_tensor = cpu_tensor.gpu()
65 self.assertAllEqual(cpu_tensor + gpu_tensor, 2.0)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | ops_test.py | 281 gpu_tensor = cpu_tensor.gpu()
282 self.assertAllEqual(cpu_tensor + gpu_tensor, 2.0)
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session_test.cc | 1939 Tensor gpu_tensor; in TEST() local
1953 gpu_tensor = outputs[0]; in TEST()
1954 ASSERT_TRUE(IsCUDATensor(gpu_tensor)); in TEST()
1963 TF_ASSERT_OK(session->RunCallable(feed_gpu_fetch_cpu, {gpu_tensor}, in TEST()
2012 Tensor gpu_tensor; in TestFeedAndFetchTensorsInDeviceMemory() local
2027 gpu_tensor = outputs[0]; in TestFeedAndFetchTensorsInDeviceMemory()
2028 ASSERT_TRUE(IsCUDATensor(gpu_tensor)) << DataType_Name(dtype); in TestFeedAndFetchTensorsInDeviceMemory()
2039 TF_ASSERT_OK(session->RunCallable(handle, {gpu_tensor}, &outputs, nullptr)) in TestFeedAndFetchTensorsInDeviceMemory()
|