Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed_thin.h35 class ConvolutionTransposedThin : public GPUOperation {
37 ConvolutionTransposedThin() = default;
41 ConvolutionTransposedThin(ConvolutionTransposedThin&& operation);
42 ConvolutionTransposedThin& operator=(ConvolutionTransposedThin&& operation);
43 ConvolutionTransposedThin(const ConvolutionTransposedThin&) = delete;
44 ConvolutionTransposedThin& operator=(const ConvolutionTransposedThin&) =
48 friend ConvolutionTransposedThin CreateConvolutionTransposedThin(
51 ConvolutionTransposedThin(const OperationDef& definition,
67 void ConvolutionTransposedThin::UploadData( in UploadData()
107 void ConvolutionTransposedThin::RearrangeWeightsData( in RearrangeWeightsData()
[all …]
Dconvolution_transposed_thin.cc27 ConvolutionTransposedThin::ConvolutionTransposedThin( in ConvolutionTransposedThin() function in tflite::gpu::ConvolutionTransposedThin
40 ConvolutionTransposedThin::ConvolutionTransposedThin( in ConvolutionTransposedThin() function in tflite::gpu::ConvolutionTransposedThin
41 ConvolutionTransposedThin&& operation) in ConvolutionTransposedThin()
44 ConvolutionTransposedThin& ConvolutionTransposedThin::operator=( in operator =()
45 ConvolutionTransposedThin&& operation) { in operator =()
52 std::string ConvolutionTransposedThin::GenerateConvolutionTransposedCode( in GenerateConvolutionTransposedCode()
150 int3 ConvolutionTransposedThin::GetGridSize() const { in GetGridSize()
165 ConvolutionTransposedThin CreateConvolutionTransposedThin( in CreateConvolutionTransposedThin()
168 ConvolutionTransposedThin result(definition, attr, gpu_info); in CreateConvolutionTransposedThin()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconvolution_transposed_thin_test.cc58 ConvolutionTransposedThin operation = CreateConvolutionTransposedThin( in TEST_F()
62 absl::make_unique<ConvolutionTransposedThin>(std::move(operation)), in TEST_F()
75 TEST_F(OpenCLOperationTest, ConvolutionTransposedThin) { in TEST_F() argument
98 ConvolutionTransposedThin operation = CreateConvolutionTransposedThin( in TEST_F()
102 absl::make_unique<ConvolutionTransposedThin>(std::move(operation)), in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_transposed_selector.cc32 ConvolutionTransposedThin conv = in SelectConvolutionTransposedAdreno()
34 return absl::make_unique<ConvolutionTransposedThin>(std::move(conv)); in SelectConvolutionTransposedAdreno()
50 ConvolutionTransposedThin conv = in SelectConvolutionTransposedPowerVR()
52 return absl::make_unique<ConvolutionTransposedThin>(std::move(conv)); in SelectConvolutionTransposedPowerVR()