Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_metal.h29 class ConvolutionMetal : public GPUOperation {
56 ConvolutionMetal() = default;
67 ConvolutionMetal(ConvolutionMetal&& kernel) = default;
68 ConvolutionMetal& operator=(ConvolutionMetal&& kernel) = default;
69 ConvolutionMetal(const ConvolutionMetal&) = delete;
70 ConvolutionMetal& operator=(const ConvolutionMetal&) = delete;
80 explicit ConvolutionMetal(const OperationDef& definition) in ConvolutionMetal() function
82 friend ConvolutionMetal CreateConvolutionMetal(
86 friend ConvolutionMetal CreateConvolutionMetalWino4x4To6x6(
95 ConvolutionMetal CreateConvolutionMetal(const OperationDef& definition,
[all …]
Dconv_metal.cc158 std::string GenerateConvolution(const ConvolutionMetal::ConvParams& params, in GenerateConvolution()
175 ConvolutionMetal::WeightsUploadType::CONSTANT_MEM in GenerateConvolution()
180 ConvolutionMetal::WeightsUploadType::LOCAL_MEM_BY_THREADS; in GenerateConvolution()
186 ConvolutionMetal::WeightsUploadType::PRIVATE_MEM_SIMD8_BROADCAST || in GenerateConvolution()
188 ConvolutionMetal::WeightsUploadType::PRIVATE_MEM_SIMD16_BROADCAST || in GenerateConvolution()
190 ConvolutionMetal::WeightsUploadType::PRIVATE_MEM_SIMD32_BROADCAST; in GenerateConvolution()
193 ConvolutionMetal::WeightsUploadType::PRIVATE_MEM_SIMD8_BROADCAST) { in GenerateConvolution()
195 } else if (params.weights_upload_type == ConvolutionMetal::WeightsUploadType:: in GenerateConvolution()
198 } else if (params.weights_upload_type == ConvolutionMetal::WeightsUploadType:: in GenerateConvolution()
691 ConvolutionMetal::ConvParams GetConvParamsForA7A8( in GetConvParamsForA7A8()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_selector.cc142 ConvolutionMetal conv = in SelectConvolution()
144 return absl::make_unique<ConvolutionMetal>(std::move(conv)); in SelectConvolution()
164 ConvolutionMetal conv = in SelectConvolutionForWinograd()
166 return absl::make_unique<ConvolutionMetal>(std::move(conv)); in SelectConvolutionForWinograd()
192 ConvolutionMetal conv = in SelectConvolutionWithDynamicWeights()
194 return absl::make_unique<ConvolutionMetal>(std::move(conv)); in SelectConvolutionWithDynamicWeights()
Dfully_connected_selector.cc97 ConvolutionMetal conv = in SelectFullyConnected()
99 return absl::make_unique<ConvolutionMetal>(std::move(conv)); in SelectFullyConnected()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dconv_test.mm67 ConvolutionMetal operation =
70 src_tensor, absl::make_unique<ConvolutionMetal>(std::move(operation)), BHWC(1, 2, 2, 2),
103 ConvolutionMetal operation =
106 src_tensor, absl::make_unique<ConvolutionMetal>(std::move(operation)), BHWC(1, 1, 1, 1),
139 ConvolutionMetal operation =
142 src_tensor, absl::make_unique<ConvolutionMetal>(std::move(operation)), BHWC(1, 1, 1, 1),
175 ConvolutionMetal operation =
178 src_tensor, absl::make_unique<ConvolutionMetal>(std::move(operation)), BHWC(1, 2, 1, 2),
211 ConvolutionMetal operation =
214 src_tensor, absl::make_unique<ConvolutionMetal>(std::move(operation)), BHWC(1, 2, 2, 1),
[all …]