Searched defs:gpu_tensor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 256 const Tensor* gpu_tensor, Tensor* cpu_tensor, in CopyGPUTensorToCPU()
301 Device* gpu_device, Tensor* gpu_tensor, in CopyCPUTensorToGPU()
|
| D | gpu_device_test.cc | 104 void CopyCPUToGPU(Tensor* cpu_tensor, Tensor* gpu_tensor, Device* device, in CopyCPUToGPU()
110 void CopyGPUToCPU(Tensor* gpu_tensor, Tensor* cpu_tensor, Device* device, in CopyGPUToCPU()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 722 const auto& gpu_tensor = *GetTensor(id); in GetOutputTensor() local
|
| D | cl_arguments.cc | 197 Tensor gpu_tensor; in CreateCLObject() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_arguments.cc | 154 MetalSpatialTensor gpu_tensor; in CreateMetalObject() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session_test.cc | 2368 Tensor gpu_tensor; in TEST() local
2441 Tensor gpu_tensor; in TestFeedAndFetchTensorsInDeviceMemory() local
|