Home
last modified time | relevance | path

Searched refs:DepthwiseConvolutionMethod (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/nodes/
H A DDepthwiseConvolutionLayerNode.h46DepthwiseConvolutionMethod method = DepthwiseConvolutionMethod::Default,
52 void set_depthwise_convolution_method(DepthwiseConvolutionMethod method);
60 DepthwiseConvolutionMethod depthwise_convolution_method() const;
112 DepthwiseConvolutionMethod _method;
H A DFusedDepthwiseConvolutionBatchNormalizationNode.h48 DepthwiseConvolutionMethod method,
55 void set_depthwise_convolution_method(DepthwiseConvolutionMethod method);
64 DepthwiseConvolutionMethod depthwise_convolution_method() const;
130 DepthwiseConvolutionMethod _method;
/aosp_15_r20/external/ComputeLibrary/src/graph/
H A DTypeLoader.cpp105 DepthwiseConvolutionMethod depthwise_convolution_method_from_name(const std::string &name) in depthwise_convolution_method_from_name()
107 static const std::map<std::string, DepthwiseConvolutionMethod> methods = in depthwise_convolution_method_from_name()
109 { "default", DepthwiseConvolutionMethod::Default }, in depthwise_convolution_method_from_name()
110 { "optimized3x3", DepthwiseConvolutionMethod::Optimized3x3 }, in depthwise_convolution_method_from_name()
H A DGraphBuilder.cpp333 … PadStrideInfo conv_info, int depth_multiplier, DepthwiseConvolutionMethod method, in add_depthwise_convolution_node()
/aosp_15_r20/external/ComputeLibrary/tests/validate_examples/
H A Dgraph_depthwiseconvolution.cpp102 const std::set<arm_compute::graph::DepthwiseConvolutionMethod> supported_convolution_methods in DepthConvolutionOptions()
104 arm_compute::graph::DepthwiseConvolutionMethod::Default, in DepthConvolutionOptions()
105 arm_compute::graph::DepthwiseConvolutionMethod::GEMV, in DepthConvolutionOptions()
106 arm_compute::graph::DepthwiseConvolutionMethod::Optimized3x3, in DepthConvolutionOptions()
116 …conv_mode = parser.add_option<EnumOption<arm_compute::graph::DepthwiseConvolutionMethod>>("conv… in DepthConvolutionOptions()
117 … arm_compute::graph::DepthwiseConvolutionMethod::Default); in DepthConvolutionOptions()
236 …EnumOption<arm_compute::graph::DepthwiseConvolutionMethod> *conv_mode; /**< Convolution m…
H A Dgraph_validate_utils.h181 …arm_compute::graph::DepthwiseConvolutionMethod depth_convolution_method{ arm_compute::graph::Depth…
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/frontend/
H A DTypes.h53 using graph::DepthwiseConvolutionMethod;
65DepthwiseConvolutionMethod depthwise_convolution_method_hint = { DepthwiseConvolutionMethod::Defau…
H A DIStreamOperators.h94 inline IStream &operator<<(IStream &s, DepthwiseConvolutionMethod depthwise_convolution_method_hint)
/aosp_15_r20/external/ComputeLibrary/src/graph/nodes/
H A DFusedDepthwiseConvolutionBatchNormalizationNode.cpp38DepthwiseConvolutionMethod method, in FusedDepthwiseConvolutionBatchNormalizationNode()
46 …volutionBatchNormalizationNode::set_depthwise_convolution_method(DepthwiseConvolutionMethod method) in set_depthwise_convolution_method()
51 DepthwiseConvolutionMethod FusedDepthwiseConvolutionBatchNormalizationNode::depthwise_convolution_m… in depthwise_convolution_method()
H A DDepthwiseConvolutionLayerNode.cpp35 …seConvolutionLayerNode(PadStrideInfo info, int depth_multiplier, DepthwiseConvolutionMethod method, in DepthwiseConvolutionLayerNode()
48 void DepthwiseConvolutionLayerNode::set_depthwise_convolution_method(DepthwiseConvolutionMethod met… in set_depthwise_convolution_method()
53 DepthwiseConvolutionMethod DepthwiseConvolutionLayerNode::depthwise_convolution_method() const in depthwise_convolution_method()
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/
H A DTypePrinter.h280 inline ::std::ostream &operator<<(::std::ostream &os, const DepthwiseConvolutionMethod &method)
284 case DepthwiseConvolutionMethod::Default:
287 case DepthwiseConvolutionMethod::Optimized3x3:
H A DTypeLoader.h110 DepthwiseConvolutionMethod depthwise_convolution_method_from_name(const std::string &name);
119 inline ::std::istream &operator>>(::std::istream &stream, DepthwiseConvolutionMethod &target)
H A DGraphBuilder.h217DepthwiseConvolutionMethod method = DepthwiseConvolutionMethod::Default,
H A DTypes.h135 enum class DepthwiseConvolutionMethod enum
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/backends/
H A DValidateHelpers.h244 const DepthwiseConvolutionMethod dwc_algorithm = node.depthwise_convolution_method(); in validate_depthwise_convolution_layer()
251 case DepthwiseConvolutionMethod::Default: in validate_depthwise_convolution_layer()
252 case DepthwiseConvolutionMethod::Optimized3x3: in validate_depthwise_convolution_layer()
/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DNodeExecutionMethodMutator.cpp98 casted_node->set_depthwise_convolution_method(DepthwiseConvolutionMethod::Default); in mutate()