Searched refs:kOHWIOGroupO4I4 (Results 1 – 8 of 8) sorted by relevance
23 layout == WeightsLayout::kOHWIOGroupO4I4 || in GetOutputGroupSize()39 return layout == WeightsLayout::kOHWIOGroupO4I4 || in IsO4I4()
23 weight_desc.layout == WeightsLayout::kOHWIOGroupO4I4 || in GetTotalElementsCountForLayout()58 } else if (dst_weight_desc.layout == WeightsLayout::kOHWIOGroupO4I4) { in RearrangeWeights()
27 kOHWIOGroupO4I4, enumerator
124 {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()
479 if (params.weights_layout == WeightsLayout::kOHWIOGroupO4I4) { in GenerateConvolution()707 params.weights_layout = WeightsLayout::kOHWIOGroupO4I4; in GetConvParamsForA7A8()808 params.weights_layout = WeightsLayout::kOHWIOGroupO4I4; in GetConvParamsForA9AndHigher()876 params.weights_layout = WeightsLayout::kOHWIOGroupO4I4; in GetConvParamsForIntel()914 params.weights_layout = WeightsLayout::kOHWIOGroupO4I4; in GetConvParamsForAMD()950 params.weights_layout = WeightsLayout::kOHWIOGroupO4I4; in GetConvParams()1090 params.weights_layout = WeightsLayout::kOHWIOGroupO4I4; in CreateConvolutionMetalWino4x4To6x6()
40 weights_layout_ = WeightsLayout::kOHWIOGroupO4I4; in ConvolutionTransposed()86 weights_layout_ = WeightsLayout::kOHWIOGroupO4I4; in ConvolutionTransposed()138 weights_layout_ == WeightsLayout::kOHWIOGroupO4I4) { in GenerateConvolutionTransposedCode()
154 conv_weights_desc.layout == WeightsLayout::kOHWIOGroupO4I4) { in GetConverterToConvWeightsCode()
1324 conv_params.weights_layout = WeightsLayout::kOHWIOGroupO4I4; in GuessBestParams()