Home
last modified time | relevance | path

Searched refs:ConverterToConvWeights (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter.h27 class ConverterToConvWeights : public GPUOperation {
29 ConverterToConvWeights(const OperationDef& definition,
35 ConverterToConvWeights(ConverterToConvWeights&& operation);
36 ConverterToConvWeights& operator=(ConverterToConvWeights&& operation);
37 ConverterToConvWeights(const ConverterToConvWeights&) = delete;
38 ConverterToConvWeights& operator=(const ConverterToConvWeights&) = delete;
51 ConverterToConvWeights CreateConverterToConvWeights(
Dconv_weights_converter.cc27 ConverterToConvWeights::ConverterToConvWeights( in ConverterToConvWeights() function in tflite::gpu::ConverterToConvWeights
33 ConverterToConvWeights::ConverterToConvWeights( in ConverterToConvWeights() function in tflite::gpu::ConverterToConvWeights
34 ConverterToConvWeights&& operation) in ConverterToConvWeights()
38 ConverterToConvWeights& ConverterToConvWeights::operator=( in operator =()
39 ConverterToConvWeights&& operation) { in operator =()
47 std::string ConverterToConvWeights::GetConverterToConvWeightsCode( in GetConverterToConvWeightsCode()
170 absl::Status ConverterToConvWeights::BindArguments(ArgumentsBinder* args) { in BindArguments()
178 int3 ConverterToConvWeights::GetGridSize() const { in GetGridSize()
187 ConverterToConvWeights CreateConverterToConvWeights( in CreateConverterToConvWeights()
189 return ConverterToConvWeights(definition, weights_desc); in CreateConverterToConvWeights()
Dconv_weights_converter_test_util.cc93 auto converter = ConverterToConvWeights(op_def, weight_desc); in ConvolutionWeightsConverterTest()
96 absl::make_unique<ConverterToConvWeights>(std::move(converter)), in ConvolutionWeightsConverterTest()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_selector.cc214 ConverterToConvWeights converter = in SelectConverterToConvWeights()
215 ConverterToConvWeights(op_def, weights_desc); in SelectConverterToConvWeights()
216 return absl::make_unique<ConverterToConvWeights>(std::move(converter)); in SelectConverterToConvWeights()