Searched refs:DepthwiseConvolutionMethod (Results 1 – 16 of 16) sorted by relevance
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/nodes/ |
H A D | DepthwiseConvolutionLayerNode.h | 46 … DepthwiseConvolutionMethod method = DepthwiseConvolutionMethod::Default, 52 void set_depthwise_convolution_method(DepthwiseConvolutionMethod method); 60 DepthwiseConvolutionMethod depthwise_convolution_method() const; 112 DepthwiseConvolutionMethod _method;
|
H A D | FusedDepthwiseConvolutionBatchNormalizationNode.h | 48 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 D | TypeLoader.cpp | 105 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 D | GraphBuilder.cpp | 333 … PadStrideInfo conv_info, int depth_multiplier, DepthwiseConvolutionMethod method, in add_depthwise_convolution_node()
|
/aosp_15_r20/external/ComputeLibrary/tests/validate_examples/ |
H A D | graph_depthwiseconvolution.cpp | 102 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 D | graph_validate_utils.h | 181 …arm_compute::graph::DepthwiseConvolutionMethod depth_convolution_method{ arm_compute::graph::Depth…
|
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/frontend/ |
H A D | Types.h | 53 using graph::DepthwiseConvolutionMethod; 65 …DepthwiseConvolutionMethod depthwise_convolution_method_hint = { DepthwiseConvolutionMethod::Defau…
|
H A D | IStreamOperators.h | 94 inline IStream &operator<<(IStream &s, DepthwiseConvolutionMethod depthwise_convolution_method_hint)
|
/aosp_15_r20/external/ComputeLibrary/src/graph/nodes/ |
H A D | FusedDepthwiseConvolutionBatchNormalizationNode.cpp | 38 … DepthwiseConvolutionMethod 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 D | DepthwiseConvolutionLayerNode.cpp | 35 …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 D | TypePrinter.h | 280 inline ::std::ostream &operator<<(::std::ostream &os, const DepthwiseConvolutionMethod &method) 284 case DepthwiseConvolutionMethod::Default: 287 case DepthwiseConvolutionMethod::Optimized3x3:
|
H A D | TypeLoader.h | 110 DepthwiseConvolutionMethod depthwise_convolution_method_from_name(const std::string &name); 119 inline ::std::istream &operator>>(::std::istream &stream, DepthwiseConvolutionMethod &target)
|
H A D | GraphBuilder.h | 217 … DepthwiseConvolutionMethod method = DepthwiseConvolutionMethod::Default,
|
H A D | Types.h | 135 enum class DepthwiseConvolutionMethod enum
|
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/backends/ |
H A D | ValidateHelpers.h | 244 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 D | NodeExecutionMethodMutator.cpp | 98 casted_node->set_depthwise_convolution_method(DepthwiseConvolutionMethod::Default); in mutate()
|