Home
last modified time | relevance | path

Searched refs:ConvolutionTransposed (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed.h39 class ConvolutionTransposed : public GPUOperation {
41 ConvolutionTransposed() = default;
50 ConvolutionTransposed(ConvolutionTransposed&& operation) = default;
51 ConvolutionTransposed& operator=(ConvolutionTransposed&& operation) = default;
52 ConvolutionTransposed(const ConvolutionTransposed&) = delete;
53 ConvolutionTransposed& operator=(const ConvolutionTransposed&) = delete;
63 friend ConvolutionTransposed CreateConvolutionTransposed(
66 friend ConvolutionTransposed CreateConvolutionTransposed3D(
69 friend ConvolutionTransposed CreateConvolutionTransposedDynamicWeights(
73 ConvolutionTransposed(const OperationDef& definition,
[all …]
Dconvolution_transposed_test_util.cc53 ConvolutionTransposed operation = in ConvolutionTransposedSimpleWeightsTest()
57 absl::make_unique<ConvolutionTransposed>(std::move(operation)), in ConvolutionTransposedSimpleWeightsTest()
93 ConvolutionTransposed operation = in ConvolutionTransposedTest()
97 absl::make_unique<ConvolutionTransposed>(std::move(operation)), in ConvolutionTransposedTest()
Dconvolution_transposed.cc32 ConvolutionTransposed::ConvolutionTransposed( in ConvolutionTransposed() function in tflite::gpu::ConvolutionTransposed
77 ConvolutionTransposed::ConvolutionTransposed( in ConvolutionTransposed() function in tflite::gpu::ConvolutionTransposed
127 std::string ConvolutionTransposed::GenerateConvolutionTransposedCode( in GenerateConvolutionTransposedCode()
550 absl::Status ConvolutionTransposed::BindArguments(ArgumentsBinder* args) { in BindArguments()
560 int3 ConvolutionTransposed::GetGridSize() const { in GetGridSize()
571 void ConvolutionTransposed::GetPossibleKernelWorkGroups( in GetPossibleKernelWorkGroups()
578 ConvolutionTransposed CreateConvolutionTransposed( in CreateConvolutionTransposed()
582 ConvolutionTransposed result(definition, attr, gpu_info, weights_are_buffer); in CreateConvolutionTransposed()
595 ConvolutionTransposed CreateConvolutionTransposed3D( in CreateConvolutionTransposed3D()
599 ConvolutionTransposed result(definition, attr, gpu_info, weights_are_buffer); in CreateConvolutionTransposed3D()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_transposed_selector.cc40 ConvolutionTransposed conv = in SelectConvolutionTransposedAdreno()
42 return absl::make_unique<ConvolutionTransposed>(std::move(conv)); in SelectConvolutionTransposedAdreno()
66 ConvolutionTransposed conv = in SelectConvolutionTransposedPowerVR()
68 return absl::make_unique<ConvolutionTransposed>(std::move(conv)); in SelectConvolutionTransposedPowerVR()
75 ConvolutionTransposed conv = in SelectConvolutionTransposedMali()
77 return absl::make_unique<ConvolutionTransposed>(std::move(conv)); in SelectConvolutionTransposedMali()
107 ConvolutionTransposed conv = in SelectConvolutionTransposedWithDynamicWeights()
110 return absl::make_unique<ConvolutionTransposed>(std::move(conv)); in SelectConvolutionTransposedWithDynamicWeights()
131 ConvolutionTransposed conv = in SelectConvolutionTransposedWithDynamicWeights()
134 return absl::make_unique<ConvolutionTransposed>(std::move(conv)); in SelectConvolutionTransposedWithDynamicWeights()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconvolution_transposed_test.cc35 TEST_F(OpenCLOperationTest, ConvolutionTransposed) { in TEST_F() argument