Home
last modified time | relevance | path

Searched refs:kOICustomSpatialI4O4 (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter.cc55 if (conv_weights_desc.layout == WeightsLayout::kOICustomSpatialI4O4 || in GetConverterToConvWeightsCode()
83 if (conv_weights_desc.layout == WeightsLayout::kOICustomSpatialI4O4 || in GetConverterToConvWeightsCode()
147 if (conv_weights_desc.layout == WeightsLayout::kOICustomSpatialI4O4 || in GetConverterToConvWeightsCode()
Dconvolution_transposed_3x3_thin.cc34 weights_layout_ = WeightsLayout::kOICustomSpatialI4O4; in ConvolutionTransposed3x3Thin()
Dconvolution_transposed_4x4.cc60 weights_layout_ = WeightsLayout::kOICustomSpatialI4O4; in ConvolutionTransposed4x4()
Dconv_weights_converter_test_util.cc258 for (auto weights_layout : {WeightsLayout::kOICustomSpatialI4O4, in ConverterToConvTransposedWeights4x4Test()
Dconvolution_transposed_3x3.cc44 weights_layout_ = WeightsLayout::kOICustomSpatialI4O4; in ConvolutionTransposed3x3()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dweights_conversion.cc29 } else if (weight_desc.layout == WeightsLayout::kOICustomSpatialI4O4 || in GetTotalElementsCountForLayout()
71 } else if (dst_weight_desc.layout == WeightsLayout::kOICustomSpatialI4O4) { in RearrangeWeights()
Dweights_layout.h28 kOICustomSpatialI4O4, enumerator
Dweights_layout.cc34 layout == WeightsLayout::kOICustomSpatialI4O4 || in IsI4O4()