Searched refs:DataFromBHWDC (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | tensor_desc.cc | 912 DataFromBHWDC(src, shape, *this, gpu_data); in UploadData()
915 DataFromBHWDC(src, shape, *this, gpu_data); in UploadData()
990 void DataFromBHWDC(const float* src, const BHWDC& shape, in DataFromBHWDC() function
1018 template void DataFromBHWDC<float>(const float* src, const BHWDC& shape,
1020 template void DataFromBHWDC<half>(const float* src, const BHWDC& shape,
|
| D | tensor_desc.h | 190 void DataFromBHWDC(const float* src, const BHWDC& shape,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_spatial_tensor.cc | 395 DataFromBHWDC(in, shape_, descriptor_, data_f.get()); in WriteDataBHWDC()
399 DataFromBHWDC(in, shape_, descriptor_, data_h.get()); in WriteDataBHWDC()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | tensor.cc | 499 DataFromBHWDC(in, shape_, descriptor_, data_f.get()); in WriteDataBHWDC()
503 DataFromBHWDC(in, shape_, descriptor_, data_h.get()); in WriteDataBHWDC()
|