Home
last modified time | relevance | path

Searched refs:kOHWIOGroupI4O4 (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dweights_layout.cc22 if (layout == WeightsLayout::kOHWIOGroupI4O4 || in GetOutputGroupSize()
33 return layout == WeightsLayout::kOHWIOGroupI4O4 || in IsI4O4()
Dweights_conversion.cc22 if (weight_desc.layout == WeightsLayout::kOHWIOGroupI4O4 || in GetTotalElementsCountForLayout()
45 if (dst_weight_desc.layout == WeightsLayout::kOHWIOGroupI4O4) { in RearrangeWeights()
Dweights_layout.h26 kOHWIOGroupI4O4, enumerator
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter_test_util.cc124 {WeightsLayout::kOHWIOGroupI4O4, WeightsLayout::kOHWIOGroupO4I4}) { in ConverterToConvWeights1x1OutX4Test()
158 {WeightsLayout::kOHWIOGroupI4O4, WeightsLayout::kOHWIOGroupO4I4}) { in ConverterToConvWeights1x1OutX4UnalignedTest()
191 {WeightsLayout::kOHWIOGroupI4O4, WeightsLayout::kOHWIOGroupO4I4}) { in ConverterToConvWeights1x1OutX2Test()
224 {WeightsLayout::kOHWIOGroupI4O4, WeightsLayout::kOHWIOGroupO4I4}) { in ConverterToConvWeightsOutX2Test()
Dconvolution_transposed.cc42 weights_layout_ = WeightsLayout::kOHWIOGroupI4O4; in ConvolutionTransposed()
88 weights_layout_ = WeightsLayout::kOHWIOGroupI4O4; in ConvolutionTransposed()
137 if (weights_layout_ == WeightsLayout::kOHWIOGroupI4O4 || in GenerateConvolutionTransposedCode()
Dconv_weights_converter.cc153 } else if (conv_weights_desc.layout == WeightsLayout::kOHWIOGroupI4O4 || in GetConverterToConvWeightsCode()
Dconv_metal.cc878 params.weights_layout = WeightsLayout::kOHWIOGroupI4O4; in GetConvParamsForIntel()
916 params.weights_layout = WeightsLayout::kOHWIOGroupI4O4; in GetConvParamsForAMD()
1103 params.weights_layout = WeightsLayout::kOHWIOGroupI4O4; in CreateConvolutionMetalWino4x4To6x6()
1109 params.weights_layout = WeightsLayout::kOHWIOGroupI4O4; in CreateConvolutionMetalWino4x4To6x6()
1115 params.weights_layout = WeightsLayout::kOHWIOGroupI4O4; in CreateConvolutionMetalWino4x4To6x6()
Dconv_buffer_1x1.h53 desc.layout = WeightsLayout::kOHWIOGroupI4O4; in GetWeightsDescription()
Dconv_powervr.cc1326 conv_params.weights_layout = WeightsLayout::kOHWIOGroupI4O4; in GuessBestParams()