Searched refs:weights_layout (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_weights_converter_test_util.cc | 123 for (auto weights_layout : in ConverterToConvWeights1x1OutX4Test()
125 conv_weight_desc.layout = weights_layout; in ConverterToConvWeights1x1OutX4Test()
157 for (auto weights_layout : in ConverterToConvWeights1x1OutX4UnalignedTest()
159 conv_weight_desc.layout = weights_layout; in ConverterToConvWeights1x1OutX4UnalignedTest()
190 for (auto weights_layout : in ConverterToConvWeights1x1OutX2Test()
192 conv_weight_desc.layout = weights_layout; in ConverterToConvWeights1x1OutX2Test()
223 for (auto weights_layout : in ConverterToConvWeightsOutX2Test()
225 conv_weight_desc.layout = weights_layout; in ConverterToConvWeightsOutX2Test()
258 for (auto weights_layout : {WeightsLayout::kOICustomSpatialI4O4, in ConverterToConvTransposedWeights4x4Test()
260 weight_desc.layout = weights_layout; in ConverterToConvTransposedWeights4x4Test()
[all …]
|
| D | conv_metal.h | 50 WeightsLayout weights_layout; member
74 desc.layout = params_.weights_layout; in GetWeightsDescription()
|
| D | conv_metal.cc | 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()
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 …]
|
| D | BUILD | 88 "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
129 "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
153 "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
181 "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
218 "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
253 "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
274 "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
295 "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
|
| D | conv_powervr.h | 52 desc.layout = conv_params_.weights_layout; in GetWeightsDescription()
95 WeightsLayout weights_layout; member
|
| D | conv_powervr.cc | 1324 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()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | BUILD | 197 ":weights_layout",
209 name = "weights_layout",
210 srcs = ["weights_layout.cc"],
211 hdrs = ["weights_layout.h"],
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | BUILD | 17 "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
36 "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | BUILD | 22 "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
34 "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
|