Searched refs:DownloadData (Results 1 – 17 of 17) sorted by relevance
94 void DownloadData(tflite::gpu::Tensor<BHWC, T>* dst);98 void DownloadData(tflite::gpu::Tensor<BHWDC, T>* dst);260 void DownloadData(T* dst);320 void TensorDescriptor::DownloadData(tflite::gpu::Tensor<BHWC, T>* dst) { in DownloadData() function323 DownloadData(dst->data.data()); in DownloadData()333 void TensorDescriptor::DownloadData(tflite::gpu::Tensor<BHWDC, T>* dst) { in DownloadData() function336 DownloadData(dst->data.data()); in DownloadData()362 void TensorDescriptor::DownloadData(T* dst) { in DownloadData() function
76 dst_cpu_descs[i].DownloadData(dst_cpu[i]); in ExecuteGPUOperation()106 dst_cpu_descs[i].DownloadData(dst_cpu[i]); in ExecuteGPUOperation()156 dst_cpu_descs[i].DownloadData(&inputs[i]); in ExecuteGpuModel()
53 dst_desc.DownloadData(&dst_tensor); in CastTests()88 dst_desc.DownloadData(&dst_tensor); in CastToBoolTests()123 dst_desc.DownloadData(&dst_tensor); in CastFromBoolTests()
46 dst.DownloadData(&dst_tensor); in OneHotTest()75 dst.DownloadData(&dst_tensor); in OneHotBatchTest()
105 dst.DownloadData(&dst_tensor); in CumsumHWC()202 dst.DownloadData(&dst_tensor); in CumsumBHWC()
56 dst.DownloadData(&dst_tensor); in TransposeIntTest()98 dst.DownloadData(&dst_tensor); in TransposeUintTest()
182 dst_0.DownloadData(&dst_tensor); in MaxPoolingIndicesTest()184 dst_1.DownloadData(&dst_tensor_ind); in MaxPoolingIndicesTest()
57 dst.DownloadData(&dst_tensor); in AddTwoEqualIntTensorsTest()101 dst.DownloadData(&dst_tensor); in AddTwoEqualUintTensorsTest()
55 dst.DownloadData(&dst_tensor); in ReduceSumChannelsIntTest()96 dst.DownloadData(&dst_tensor); in ReduceProductChannelsUIntTest()
58 dst.DownloadData(&dst_tensor); in GatherWidthIntTest()
1020 dst_desc.DownloadData(&dst_tensor); in LessTest()1059 dst_desc.DownloadData(&dst_tensor); in LessEqualTest()1098 dst_desc.DownloadData(&dst_tensor); in GreaterTest()1137 dst_desc.DownloadData(&dst_tensor); in GreaterEqualTest()1176 dst_desc.DownloadData(&dst_tensor); in EqualTest()1215 dst_desc.DownloadData(&dst_tensor); in NotEqualTest()
136 dst_descriptor.DownloadData(&dst_tensor); in RunSelectV2()
70 output_descriptor.DownloadData(&tensor_gpu); in TensorBHWCTest()117 output_descriptor.DownloadData(&tensor_gpu); in TensorBHWDCTest()
923 desc.DownloadData(result); in GetOutputTensor()
82 output_descriptor.DownloadData(&tensor_gpu);163 output_descriptor.DownloadData(&tensor_gpu);
364 desc.DownloadData(result); in GetOutputTensor()
123 desc.DownloadData(&cpu_tensor); in RunExternalImmutableSample()