Searched defs:gpu_tensor (Results 1 – 2 of 2) sorted by relevance
257 const Tensor* gpu_tensor, Tensor* cpu_tensor, in CopyGPUTensorToCPU()302 Device* gpu_device, Tensor* gpu_tensor, in CopyCPUTensorToGPU()
1939 Tensor gpu_tensor; in TEST() local2012 Tensor gpu_tensor; in TestFeedAndFetchTensorsInDeviceMemory() local