| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 256 const Tensor* gpu_tensor, Tensor* cpu_tensor, in CopyGPUTensorToCPU() argument
261 Status s = PrepareCopy(gpu_device, device_context, *gpu_tensor, cpu_tensor, in CopyGPUTensorToCPU()
278 const int64 total_bytes = gpu_tensor->TotalBytes(); in CopyGPUTensorToCPU()
280 void* src_ptr = GetBase(gpu_tensor); in CopyGPUTensorToCPU()
286 TensorReference input_ref(*gpu_tensor); in CopyGPUTensorToCPU()
301 Device* gpu_device, Tensor* gpu_tensor, in CopyCPUTensorToGPU() argument
306 Status s = PrepareCopy(gpu_device, device_context, *cpu_tensor, gpu_tensor, in CopyCPUTensorToGPU()
329 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,
|
| D | gpu_device_test.cc | 104 void CopyCPUToGPU(Tensor* cpu_tensor, Tensor* gpu_tensor, Device* device, in CopyCPUToGPU() argument
107 gpu_tensor)); in CopyCPUToGPU()
110 void CopyGPUToCPU(Tensor* gpu_tensor, Tensor* cpu_tensor, Device* device, in CopyGPUToCPU() argument
113 gpu_tensor, /*tensor_name=*/"", device, cpu_tensor)); in CopyGPUToCPU()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | BUILD | 70 "//tensorflow/lite/delegates/gpu/common/task:gpu_tensor",
80 name = "gpu_tensor",
81 hdrs = ["gpu_tensor.h"],
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 722 const auto& gpu_tensor = *GetTensor(id); in GetOutputTensor() local
723 const auto dst_shape = BHWC(gpu_tensor.Batch(), gpu_tensor.Height(), in GetOutputTensor()
724 gpu_tensor.Width(), gpu_tensor.Channels()); in GetOutputTensor()
728 return gpu_tensor.ReadData(queue, result); in GetOutputTensor()
|
| D | cl_arguments.cc | 197 Tensor gpu_tensor; in CreateCLObject() local
198 RETURN_IF_ERROR(gpu_tensor.CreateFromDescriptor(*tensor_desc, context)); in CreateCLObject()
199 *result = absl::make_unique<Tensor>(std::move(gpu_tensor)); in CreateCLObject()
|
| D | BUILD | 516 "//tensorflow/lite/delegates/gpu/common/task:gpu_tensor",
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_arguments.cc | 154 MetalSpatialTensor gpu_tensor; in CreateMetalObject() local
155 RETURN_IF_ERROR(gpu_tensor.CreateFromDescriptor(*tensor_desc, device)); in CreateMetalObject()
156 *result = absl::make_unique<MetalSpatialTensor>(std::move(gpu_tensor)); in CreateMetalObject()
|
| D | BUILD | 246 "//tensorflow/lite/delegates/gpu/common/task:gpu_tensor",
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | ops_test.py | 291 gpu_tensor = cpu_tensor.gpu()
292 self.assertAllEqual(cpu_tensor + gpu_tensor, 2.0)
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session_test.cc | 2368 Tensor gpu_tensor; in TEST() local
2382 gpu_tensor = outputs[0]; in TEST()
2383 ASSERT_TRUE(IsCUDATensor(gpu_tensor)); in TEST()
2392 TF_ASSERT_OK(session->RunCallable(feed_gpu_fetch_cpu, {gpu_tensor}, in TEST()
2441 Tensor gpu_tensor; in TestFeedAndFetchTensorsInDeviceMemory() local
2456 gpu_tensor = outputs[0]; in TestFeedAndFetchTensorsInDeviceMemory()
2457 ASSERT_TRUE(IsCUDATensor(gpu_tensor)) << DataType_Name(dtype); in TestFeedAndFetchTensorsInDeviceMemory()
2468 TF_ASSERT_OK(session->RunCallable(handle, {gpu_tensor}, &outputs, nullptr)) in TestFeedAndFetchTensorsInDeviceMemory()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 73 gpu_tensor = cpu_tensor.gpu()
74 self.assertAllEqual(cpu_tensor + gpu_tensor, 2.0)
|