Home
last modified time | relevance | path

Searched refs:WeightsDescription (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter.h30 const WeightsDescription& weights_desc);
42 const OperationDef& op_def, const WeightsDescription& weights_desc);
44 WeightsDescription weights_desc_;
52 const OperationDef& definition, const WeightsDescription& weights_desc);
Dconv_weights_converter_test_util.cc31 const WeightsDescription& weight_desc, TestExecutionEnvironment* env, in ConvolutionWeightsConverterTest()
111 WeightsDescription conv_weight_desc; in ConverterToConvWeights1x1OutX4Test()
145 WeightsDescription conv_weight_desc; in ConverterToConvWeights1x1OutX4UnalignedTest()
178 WeightsDescription conv_weight_desc; in ConverterToConvWeights1x1OutX2Test()
211 WeightsDescription conv_weight_desc; in ConverterToConvWeightsOutX2Test()
245 WeightsDescription weight_desc; in ConverterToConvTransposedWeights4x4Test()
280 WeightsDescription conv_weight_desc; in ConverterToConvWeights4xTexturesTest()
Dconvolution_transposed_3x3_thin.h51 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription()
52 WeightsDescription desc; in GetWeightsDescription()
Dconv_weights_converter.cc28 const OperationDef& definition, const WeightsDescription& weights_desc) in ConverterToConvWeights()
48 const OperationDef& op_def, const WeightsDescription& conv_weights_desc) { in GetConverterToConvWeightsCode()
188 const OperationDef& definition, const WeightsDescription& weights_desc) { in CreateConverterToConvWeights()
Dconvolution_transposed_3x3.h54 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription()
55 WeightsDescription desc; in GetWeightsDescription()
Dconvolution_transposed_4x4.h56 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription()
57 WeightsDescription desc; in GetWeightsDescription()
Dconv_metal.h72 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription()
73 WeightsDescription desc; in GetWeightsDescription()
Dconv_buffer_1x1.h51 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription()
52 WeightsDescription desc; in GetWeightsDescription()
Dconvolution_transposed.h55 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription()
56 WeightsDescription desc; in GetWeightsDescription()
Dconv_powervr.h50 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription()
51 WeightsDescription desc; in GetWeightsDescription()
Dconv_metal.cc594 const WeightsDescription& weights_desc, const DataType& weights_type) { in ReorderWeightsForConv()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dweights_layout.cc21 int WeightsDescription::GetOutputGroupSize() const { in GetOutputGroupSize()
32 bool WeightsDescription::IsI4O4() const { in IsI4O4()
38 bool WeightsDescription::IsO4I4() const { in IsO4I4()
Dweights_conversion.cc20 uint GetTotalElementsCountForLayout(const WeightsDescription& weight_desc, in GetTotalElementsCountForLayout()
41 const WeightsDescription& dst_weight_desc, DataType dst_type, in RearrangeWeights()
Dweights_layout.h34 struct WeightsDescription { struct
Dweights_conversion.h327 uint GetTotalElementsCountForLayout(const WeightsDescription& weight_desc,
332 const WeightsDescription& dst_weight_desc, DataType dst_type,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dconvolution_selector.h42 ModelHints hints, WeightsDescription* weights_desc);
45 const WeightsDescription& weights_desc, const OperationDef& op_def,
Dconvolution_transposed_selector.h35 const OperationDef& op_def, WeightsDescription* weights_desc);
Doperation_selector.cc227 WeightsDescription conv_weights_desc; in GPUOperationFromNode()
304 WeightsDescription conv_weights_desc; in GPUOperationFromNode()
348 WeightsDescription weights_desc; in GPUOperationFromNode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_selector.cc62 WeightsDescription* weights_desc) { in SelectConvolutionDynamicWeightsAdreno()
120 WeightsDescription* weights_desc) { in SelectConvolutionDynamicWeightsMali()
187 WeightsDescription* weights_desc) { in SelectConvolutionWithDynamicWeights()
212 const WeightsDescription& weights_desc, const OperationDef& op_def, in SelectConverterToConvWeights()
Dconvolution_transposed_selector.cc98 const OperationDef& op_def, WeightsDescription* weights_desc) { in SelectConvolutionTransposedWithDynamicWeights()