Home
last modified time | relevance | path

Searched refs:kOICustomSpatialO4I4 (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter.cc56 conv_weights_desc.layout == WeightsLayout::kOICustomSpatialO4I4) { in GetConverterToConvWeightsCode()
84 conv_weights_desc.layout == WeightsLayout::kOICustomSpatialO4I4) { in GetConverterToConvWeightsCode()
148 conv_weights_desc.layout == WeightsLayout::kOICustomSpatialO4I4) { in GetConverterToConvWeightsCode()
Dconvolution_transposed_3x3_thin.cc32 weights_layout_ = WeightsLayout::kOICustomSpatialO4I4; in ConvolutionTransposed3x3Thin()
Dconvolution_transposed_4x4.cc58 weights_layout_ = WeightsLayout::kOICustomSpatialO4I4; in ConvolutionTransposed4x4()
Dconv_weights_converter_test_util.cc259 WeightsLayout::kOICustomSpatialO4I4}) { in ConverterToConvTransposedWeights4x4Test()
Dconvolution_transposed_3x3.cc42 weights_layout_ = WeightsLayout::kOICustomSpatialO4I4; in ConvolutionTransposed3x3()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dweights_conversion.cc30 weight_desc.layout == WeightsLayout::kOICustomSpatialO4I4) { in GetTotalElementsCountForLayout()
84 } else if (dst_weight_desc.layout == WeightsLayout::kOICustomSpatialO4I4) { in RearrangeWeights()
Dweights_layout.h29 kOICustomSpatialO4I4, enumerator
Dweights_layout.cc40 layout == WeightsLayout::kOICustomSpatialO4I4 || in IsO4I4()