Home
last modified time | relevance | path

Searched refs:WeightsLayout (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dweights_layout.cc22 if (layout == WeightsLayout::kOHWIOGroupI4O4 || in GetOutputGroupSize()
23 layout == WeightsLayout::kOHWIOGroupO4I4 || in GetOutputGroupSize()
24 layout == WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4 || in GetOutputGroupSize()
25 layout == WeightsLayout::k2DX4O4YIsHWIAndXIsOOGroupI4) { in GetOutputGroupSize()
33 return layout == WeightsLayout::kOHWIOGroupI4O4 || in IsI4O4()
34 layout == WeightsLayout::kOICustomSpatialI4O4 || in IsI4O4()
35 layout == WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4; in IsI4O4()
39 return layout == WeightsLayout::kOHWIOGroupO4I4 || in IsO4I4()
40 layout == WeightsLayout::kOICustomSpatialO4I4 || in IsO4I4()
41 layout == WeightsLayout::k2DX4O4YIsHWIAndXIsOOGroupI4; in IsO4I4()
Dweights_conversion.cc22 if (weight_desc.layout == WeightsLayout::kOHWIOGroupI4O4 || in GetTotalElementsCountForLayout()
23 weight_desc.layout == WeightsLayout::kOHWIOGroupO4I4 || in GetTotalElementsCountForLayout()
24 weight_desc.layout == WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4 || in GetTotalElementsCountForLayout()
25 weight_desc.layout == WeightsLayout::k2DX4O4YIsHWIAndXIsOOGroupI4) { in GetTotalElementsCountForLayout()
29 } else if (weight_desc.layout == WeightsLayout::kOICustomSpatialI4O4 || in GetTotalElementsCountForLayout()
30 weight_desc.layout == WeightsLayout::kOICustomSpatialO4I4) { in GetTotalElementsCountForLayout()
45 if (dst_weight_desc.layout == WeightsLayout::kOHWIOGroupI4O4) { in RearrangeWeights()
58 } else if (dst_weight_desc.layout == WeightsLayout::kOHWIOGroupO4I4) { in RearrangeWeights()
71 } else if (dst_weight_desc.layout == WeightsLayout::kOICustomSpatialI4O4) { in RearrangeWeights()
84 } else if (dst_weight_desc.layout == WeightsLayout::kOICustomSpatialO4I4) { in RearrangeWeights()
[all …]
Dweights_layout.h24 enum class WeightsLayout { enum
35 WeightsLayout layout;
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter.cc55 if (conv_weights_desc.layout == WeightsLayout::kOICustomSpatialI4O4 || in GetConverterToConvWeightsCode()
56 conv_weights_desc.layout == WeightsLayout::kOICustomSpatialO4I4) { in GetConverterToConvWeightsCode()
83 if (conv_weights_desc.layout == WeightsLayout::kOICustomSpatialI4O4 || in GetConverterToConvWeightsCode()
84 conv_weights_desc.layout == WeightsLayout::kOICustomSpatialO4I4) { in GetConverterToConvWeightsCode()
125 if (conv_weights_desc.layout == WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4 || in GetConverterToConvWeightsCode()
126 conv_weights_desc.layout == WeightsLayout::k2DX4O4YIsHWIAndXIsOOGroupI4) { in GetConverterToConvWeightsCode()
147 if (conv_weights_desc.layout == WeightsLayout::kOICustomSpatialI4O4 || in GetConverterToConvWeightsCode()
148 conv_weights_desc.layout == WeightsLayout::kOICustomSpatialO4I4) { in GetConverterToConvWeightsCode()
153 } else if (conv_weights_desc.layout == WeightsLayout::kOHWIOGroupI4O4 || in GetConverterToConvWeightsCode()
154 conv_weights_desc.layout == WeightsLayout::kOHWIOGroupO4I4) { in GetConverterToConvWeightsCode()
Dconv_weights_converter_test_util.cc60 if (weight_desc.layout == WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4 || in ConvolutionWeightsConverterTest()
61 weight_desc.layout == WeightsLayout::k2DX4O4YIsHWIAndXIsOOGroupI4) { in ConvolutionWeightsConverterTest()
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()
258 for (auto weights_layout : {WeightsLayout::kOICustomSpatialI4O4, in ConverterToConvTransposedWeights4x4Test()
259 WeightsLayout::kOICustomSpatialO4I4}) { in ConverterToConvTransposedWeights4x4Test()
293 {WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4, in ConverterToConvWeights4xTexturesTest()
294 WeightsLayout::k2DX4O4YIsHWIAndXIsOOGroupI4}) { in ConverterToConvWeights4xTexturesTest()
Dconvolution_transposed.cc40 weights_layout_ = WeightsLayout::kOHWIOGroupO4I4; in ConvolutionTransposed()
42 weights_layout_ = WeightsLayout::kOHWIOGroupI4O4; in ConvolutionTransposed()
46 weights_layout_ = WeightsLayout::k2DX4O4YIsHWIAndXIsOOGroupI4; in ConvolutionTransposed()
48 weights_layout_ = WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4; in ConvolutionTransposed()
86 weights_layout_ = WeightsLayout::kOHWIOGroupO4I4; in ConvolutionTransposed()
88 weights_layout_ = WeightsLayout::kOHWIOGroupI4O4; in ConvolutionTransposed()
92 weights_layout_ = WeightsLayout::k2DX4O4YIsHWIAndXIsOOGroupI4; in ConvolutionTransposed()
94 weights_layout_ = WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4; in ConvolutionTransposed()
137 if (weights_layout_ == WeightsLayout::kOHWIOGroupI4O4 || in GenerateConvolutionTransposedCode()
138 weights_layout_ == WeightsLayout::kOHWIOGroupO4I4) { in GenerateConvolutionTransposedCode()
Dconv_metal.cc479 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()
878 params.weights_layout = WeightsLayout::kOHWIOGroupI4O4; in GetConvParamsForIntel()
914 params.weights_layout = WeightsLayout::kOHWIOGroupO4I4; in GetConvParamsForAMD()
916 params.weights_layout = WeightsLayout::kOHWIOGroupI4O4; in GetConvParamsForAMD()
950 params.weights_layout = WeightsLayout::kOHWIOGroupO4I4; in GetConvParams()
1090 params.weights_layout = WeightsLayout::kOHWIOGroupO4I4; in CreateConvolutionMetalWino4x4To6x6()
1103 params.weights_layout = WeightsLayout::kOHWIOGroupI4O4; in CreateConvolutionMetalWino4x4To6x6()
[all …]
Dconvolution_transposed_3x3_thin.h79 WeightsLayout weights_layout_;
Dconvolution_transposed_3x3_thin.cc32 weights_layout_ = WeightsLayout::kOICustomSpatialO4I4; in ConvolutionTransposed3x3Thin()
34 weights_layout_ = WeightsLayout::kOICustomSpatialI4O4; in ConvolutionTransposed3x3Thin()
Dconvolution_transposed_3x3.h90 WeightsLayout weights_layout_;
Dconvolution_transposed_4x4.h91 WeightsLayout weights_layout_;
Dconv_metal.h50 WeightsLayout weights_layout;
Dconvolution_transposed_4x4.cc58 weights_layout_ = WeightsLayout::kOICustomSpatialO4I4; in ConvolutionTransposed4x4()
60 weights_layout_ = WeightsLayout::kOICustomSpatialI4O4; in ConvolutionTransposed4x4()
Dconvolution_transposed_3x3.cc42 weights_layout_ = WeightsLayout::kOICustomSpatialO4I4; in ConvolutionTransposed3x3()
44 weights_layout_ = WeightsLayout::kOICustomSpatialI4O4; in ConvolutionTransposed3x3()
Dconv_buffer_1x1.h53 desc.layout = WeightsLayout::kOHWIOGroupI4O4; in GetWeightsDescription()
Dconvolution_transposed.h94 WeightsLayout weights_layout_;
Dconv_powervr.cc1324 conv_params.weights_layout = WeightsLayout::kOHWIOGroupO4I4; in GuessBestParams()
1326 conv_params.weights_layout = WeightsLayout::kOHWIOGroupI4O4; in GuessBestParams()
1330 conv_params.weights_layout = WeightsLayout::k2DX4O4YIsHWIAndXIsOOGroupI4; in GuessBestParams()
1332 conv_params.weights_layout = WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4; in GuessBestParams()
Dconv_powervr.h95 WeightsLayout weights_layout;
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc359 WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4 || in GPUOperationFromNode()
361 WeightsLayout::k2DX4O4YIsHWIAndXIsOOGroupI4) { in GPUOperationFromNode()