Searched defs:OHWDI (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | weights_conversion.h | 75 const tflite::gpu::Tensor<OHWDI, S>& weights, int out_group_size, in RearrangeWeightsToODHWIOGroupI4O4()
149 const tflite::gpu::Tensor<OHWDI, S>& weights, int out_group_size, in RearrangeWeightsToODHWIOGroupO4I4()
223 const tflite::gpu::Tensor<OHWDI, S>& weights, int out_group_size, in RearrangeWeightsToI4DHWIOOGroupO4()
297 const tflite::gpu::Tensor<OHWDI, S>& weights, int out_group_size, in RearrangeWeightsToO4DHWIOOGroupI4()
371 const tflite::gpu::Tensor<OHWDI, S>& weights, in RearrangeWeightsToOICustomSpatialI4O4()
449 const tflite::gpu::Tensor<OHWDI, S>& weights, in RearrangeWeightsToOICustomSpatialO4I4()
|
| D | weights_conversion.cc | 149 const tflite::gpu::Tensor<OHWDI, DataType::FLOAT32>& weights, in RearrangeWeights()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | shape.h | 62 OHWDI = 14, enumerator
209 using OHWDI = StrongShape<Layout::OHWDI>; variable
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | convolution_transposed.h | 132 const tflite::gpu::Tensor<OHWDI, T>& weights, bool weights_are_buffer) { in UploadWeights()
|
| D | depthwise_conv.h | 185 void RearrangeWeightsForDWConv3D(const tflite::gpu::Tensor<OHWDI, S>& weights, in RearrangeWeightsForDWConv3D()
218 const tflite::gpu::Tensor<OHWDI, T>& weights, bool weights_are_buffer) { in UploadWeightsForDWConv3D()
|
| D | conv_generic.h | 297 void ConvGeneric::UploadWeights(const tflite::gpu::Tensor<OHWDI, T>& weights) { in UploadWeights()
|