Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed_4x4.h37 class ConvolutionTransposed4x4 : public GPUOperation {
39 ConvolutionTransposed4x4() = default;
50 ConvolutionTransposed4x4(ConvolutionTransposed4x4&& operation) = default;
51 ConvolutionTransposed4x4& operator=(ConvolutionTransposed4x4&& operation) =
53 ConvolutionTransposed4x4(const ConvolutionTransposed4x4&) = delete;
54 ConvolutionTransposed4x4& operator=(const ConvolutionTransposed4x4&) = delete;
71 ConvolutionTransposed4x4(const OperationDef& definition,
74 friend ConvolutionTransposed4x4 CreateConvolutionTransposed4x4(
77 friend ConvolutionTransposed4x4 CreateConvolutionTransposed4x4DynamicWeights(
98 ConvolutionTransposed4x4 CreateConvolutionTransposed4x4(
[all …]
Dconvolution_transposed_4x4.cc28 ConvolutionTransposed4x4::WeightsUploadType GetBestWeightsUploadType( in GetBestWeightsUploadType()
30 ConvolutionTransposed4x4::WeightsUploadType weights_upload_type = in GetBestWeightsUploadType()
31 ConvolutionTransposed4x4::WeightsUploadType::GLOBAL_MEM; in GetBestWeightsUploadType()
34 ConvolutionTransposed4x4::WeightsUploadType::LOCAL_MEM_ASYNC; in GetBestWeightsUploadType()
37 ConvolutionTransposed4x4::WeightsUploadType::LOCAL_MEM_BY_THREADS; in GetBestWeightsUploadType()
40 ConvolutionTransposed4x4::WeightsUploadType::CONSTANT_MEM; in GetBestWeightsUploadType()
43 ConvolutionTransposed4x4::WeightsUploadType::GLOBAL_MEM; in GetBestWeightsUploadType()
49 ConvolutionTransposed4x4::ConvolutionTransposed4x4( in ConvolutionTransposed4x4() function in tflite::gpu::ConvolutionTransposed4x4
71 std::string ConvolutionTransposed4x4::GenerateConvolutionTransposedCode( in GenerateConvolutionTransposedCode()
94 ConvolutionTransposed4x4::WeightsUploadType::CONSTANT_MEM in GenerateConvolutionTransposedCode()
[all …]
Dconvolution_transposed_4x4_test_util.cc53 ConvolutionTransposed4x4 operation = in ConvolutionTransposed4x4SimpleWeightsTest()
57 absl::make_unique<ConvolutionTransposed4x4>(std::move(operation)), in ConvolutionTransposed4x4SimpleWeightsTest()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_transposed_selector.cc62 ConvolutionTransposed4x4 conv = in SelectConvolutionTransposedPowerVR()
64 return absl::make_unique<ConvolutionTransposed4x4>(std::move(conv)); in SelectConvolutionTransposedPowerVR()
115 ConvolutionTransposed4x4 conv = in SelectConvolutionTransposedWithDynamicWeights()
118 return absl::make_unique<ConvolutionTransposed4x4>(std::move(conv)); in SelectConvolutionTransposedWithDynamicWeights()