Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed_3x3.h37 class ConvolutionTransposed3x3 : public GPUOperation {
39 ConvolutionTransposed3x3() = default;
48 ConvolutionTransposed3x3(ConvolutionTransposed3x3&& operation) = default;
49 ConvolutionTransposed3x3& operator=(ConvolutionTransposed3x3&& operation) =
51 ConvolutionTransposed3x3(const ConvolutionTransposed3x3&) = delete;
52 ConvolutionTransposed3x3& operator=(const ConvolutionTransposed3x3&) = delete;
69 ConvolutionTransposed3x3(const OperationDef& definition,
71 friend ConvolutionTransposed3x3 CreateConvolutionTransposed3x3(
74 friend ConvolutionTransposed3x3 CreateConvolutionTransposed3x3DynamicWeights(
85 ConvolutionTransposed3x3::WeightsUploadType weights_upload_type,
[all …]
Dconvolution_transposed_3x3.cc27 ConvolutionTransposed3x3::ConvolutionTransposed3x3( in ConvolutionTransposed3x3() function in tflite::gpu::ConvolutionTransposed3x3
55 std::string ConvolutionTransposed3x3::GenerateConvolutionTransposedCode( in GenerateConvolutionTransposedCode()
57 ConvolutionTransposed3x3::WeightsUploadType weights_upload_type, in GenerateConvolutionTransposedCode()
79 ConvolutionTransposed3x3::WeightsUploadType::CONSTANT_MEM in GenerateConvolutionTransposedCode()
91 ConvolutionTransposed3x3::WeightsUploadType::LOCAL_MEM_BY_THREADS || in GenerateConvolutionTransposedCode()
93 ConvolutionTransposed3x3::WeightsUploadType::LOCAL_MEM_ASYNC; in GenerateConvolutionTransposedCode()
130 ConvolutionTransposed3x3::WeightsUploadType::CONSTANT_MEM in GenerateConvolutionTransposedCode()
182 ConvolutionTransposed3x3::WeightsUploadType::LOCAL_MEM_BY_THREADS) { in GenerateConvolutionTransposedCode()
286 ConvolutionTransposed3x3::WeightsUploadType::LOCAL_MEM_ASYNC) { in GenerateConvolutionTransposedCode()
291 ConvolutionTransposed3x3::WeightsUploadType:: in GenerateConvolutionTransposedCode()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconvolution_transposed_3x3_test.cc34 TEST_F(OpenCLOperationTest, ConvolutionTransposed3x3) { in TEST_F() argument
57 ConvolutionTransposed3x3 operation = CreateConvolutionTransposed3x3( in TEST_F()
61 absl::make_unique<ConvolutionTransposed3x3>(std::move(operation)), in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_transposed_selector.cc58 ConvolutionTransposed3x3 conv = in SelectConvolutionTransposedPowerVR()
60 return absl::make_unique<ConvolutionTransposed3x3>(std::move(conv)); in SelectConvolutionTransposedPowerVR()
126 ConvolutionTransposed3x3 conv = in SelectConvolutionTransposedWithDynamicWeights()
129 return absl::make_unique<ConvolutionTransposed3x3>(std::move(conv)); in SelectConvolutionTransposedWithDynamicWeights()