Home
last modified time | relevance | path

Searched refs:weight_desc (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dweights_conversion.cc20 uint GetTotalElementsCountForLayout(const WeightsDescription& weight_desc, in GetTotalElementsCountForLayout() argument
22 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()
27 uint o_aligned = AlignByN(shape.o, 4 * weight_desc.output_group_size); in GetTotalElementsCountForLayout()
29 } else if (weight_desc.layout == WeightsLayout::kOICustomSpatialI4O4 || in GetTotalElementsCountForLayout()
30 weight_desc.layout == WeightsLayout::kOICustomSpatialO4I4) { in GetTotalElementsCountForLayout()
33 return i_aligned * o_aligned * weight_desc.spatial_remap.size(); in GetTotalElementsCountForLayout()
Dweights_conversion.h327 uint GetTotalElementsCountForLayout(const WeightsDescription& weight_desc,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter_test_util.cc31 const WeightsDescription& weight_desc, TestExecutionEnvironment* env, in ConvolutionWeightsConverterTest() argument
52 GetTotalElementsCountForLayout(weight_desc, weights.shape); in ConvolutionWeightsConverterTest()
56 RearrangeWeights(weights, weight_desc, weights_type, in ConvolutionWeightsConverterTest()
60 if (weight_desc.layout == WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4 || in ConvolutionWeightsConverterTest()
61 weight_desc.layout == WeightsLayout::k2DX4O4YIsHWIAndXIsOOGroupI4) { in ConvolutionWeightsConverterTest()
64 weight_desc.output_group_size); in ConvolutionWeightsConverterTest()
93 auto converter = ConverterToConvWeights(op_def, weight_desc); in ConvolutionWeightsConverterTest()
245 WeightsDescription weight_desc; in ConverterToConvTransposedWeights4x4Test() local
246 weight_desc.spatial_remap = {10, 11, 14, 15, 8, 9, 12, 13, in ConverterToConvTransposedWeights4x4Test()
260 weight_desc.layout = weights_layout; in ConverterToConvTransposedWeights4x4Test()
[all …]