Searched refs:DataToBHWDC (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | tensor_desc.h | 194 void DataToBHWDC(const T* src, const BHWDC& shape, const TensorDescriptor& desc,
|
| D | tensor_desc.cc | 1024 void DataToBHWDC(const T* src, const BHWDC& shape, const TensorDescriptor& desc, in DataToBHWDC() function
1048 template void DataToBHWDC<float>(const float* src, const BHWDC& shape,
1050 template void DataToBHWDC<half>(const half* src, const BHWDC& shape,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_spatial_tensor.cc | 485 DataToBHWDC(data_f.get(), shape_, descriptor_, out); in ReadDataBHWDC()
487 DataToBHWDC(data_h.get(), shape_, descriptor_, out); in ReadDataBHWDC()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | tensor.cc | 582 DataToBHWDC(data_f.get(), shape_, descriptor_, out); in ReadDataBHWDC()
584 DataToBHWDC(data_h.get(), shape_, descriptor_, out); in ReadDataBHWDC()
|