| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | weights_conversion.cc | 21 const OHWDI& shape) { in GetTotalElementsCountForLayout()
41 const OHWDI ohwdi_shape = OHWDI(shape.o, shape.h, shape.w, 1, shape.i); in GetTotalElementsCountForLayout()
47 const OHWDI ohwdi_shape = OHWDI(shape.o, shape.h, shape.w, 1, shape.i); in Get2dResourceSize()
52 const OHWDI& shape) { in Get2dResourceSize()
149 const tflite::gpu::Tensor<OHWDI, DataType::FLOAT32>& weights, in RearrangeWeights() argument
|
| D | weights_conversion.h | 75 const tflite::gpu::Tensor<OHWDI, S>& weights, int out_group_size, in RearrangeWeightsToODHWIOGroupI4O4() argument
149 const tflite::gpu::Tensor<OHWDI, S>& weights, int out_group_size, in RearrangeWeightsToODHWIOGroupO4I4() argument
223 const tflite::gpu::Tensor<OHWDI, S>& weights, int out_group_size, in RearrangeWeightsToI4DHWIOOGroupO4() argument
297 const tflite::gpu::Tensor<OHWDI, S>& weights, int out_group_size, in RearrangeWeightsToO4DHWIOOGroupI4() argument
371 const tflite::gpu::Tensor<OHWDI, S>& weights, in RearrangeWeightsToOICustomSpatialI4O4() argument
449 const tflite::gpu::Tensor<OHWDI, S>& weights, in RearrangeWeightsToOICustomSpatialO4I4() argument
492 const OHWDI& shape);
503 const OHWDI& shape);
510 const tflite::gpu::Tensor<OHWDI, DataType::FLOAT32>& weights,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | shape.h | 62 OHWDI = 14, enumerator
209 using OHWDI = StrongShape<Layout::OHWDI>; variable
376 TFLITE_GPU_LAYOUT_TRAITS(OHWDI, Axis::OUTPUT_CHANNELS, Axis::HEIGHT,
623 case Layout::OHWDI:
624 return f.template operator()<Layout::OHWDI>();
|
| D | shape_test.cc | 96 EXPECT_TRUE(HasAxis<Layout::OHWDI>(Axis::DEPTH)); in TEST()
97 EXPECT_FALSE(HasAxis<Layout::OHWDI>(Axis::CHANNELS)); in TEST()
|
| D | shape.cc | 106 case Layout::OHWDI: in ToString()
|
| D | operations.h | 294 Tensor<OHWDI, DataType::FLOAT32> weights;
345 Tensor<OHWDI, DataType::FLOAT32> weights;
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | depthwise_conv.h | 110 void UploadWeightsForDWConv3D(const tflite::gpu::Tensor<OHWDI, T>& weights,
185 void RearrangeWeightsForDWConv3D(const tflite::gpu::Tensor<OHWDI, S>& weights, in RearrangeWeightsForDWConv3D() argument
218 const tflite::gpu::Tensor<OHWDI, T>& weights, bool weights_are_buffer) { in UploadWeightsForDWConv3D() argument
|
| D | convolution_transposed.h | 88 void UploadWeights(const tflite::gpu::Tensor<OHWDI, T>& weights,
132 const tflite::gpu::Tensor<OHWDI, T>& weights, bool weights_are_buffer) { in UploadWeights() argument
|
| D | conv_generic.h | 136 void UploadWeights(const tflite::gpu::Tensor<OHWDI, T>& weights);
297 void ConvGeneric::UploadWeights(const tflite::gpu::Tensor<OHWDI, T>& weights) { in UploadWeights() argument
|