Home
last modified time | relevance | path

Searched refs:ConvolutionMethod (Results 1 – 25 of 38) sorted by relevance

12

/external/ComputeLibrary/src/runtime/NEON/functions/
DNEConvolutionLayer.cpp62 case ConvolutionMethod::WINOGRAD: in configure()
69 case ConvolutionMethod::GEMM: in configure()
76 case ConvolutionMethod::GEMM_CONV2D: in configure()
83 case ConvolutionMethod::DIRECT: in configure()
90 case ConvolutionMethod::FFT: in configure()
111 case ConvolutionMethod::WINOGRAD: in validate()
114 case ConvolutionMethod::GEMM: in validate()
117 case ConvolutionMethod::GEMM_CONV2D: in validate()
120 case ConvolutionMethod::DIRECT: in validate()
123 case ConvolutionMethod::FFT: in validate()
[all …]
/external/ComputeLibrary/src/runtime/CL/functions/
DCLConvolutionLayer.cpp66 case ConvolutionMethod::WINOGRAD: in configure()
74 case ConvolutionMethod::DIRECT: in configure()
82 case ConvolutionMethod::GEMM: in configure()
89 case ConvolutionMethod::FFT: in configure()
112 case ConvolutionMethod::WINOGRAD: in validate()
119 case ConvolutionMethod::DIRECT: in validate()
126 case ConvolutionMethod::GEMM: in validate()
132 case ConvolutionMethod::FFT: in validate()
146 ConvolutionMethod CLConvolutionLayer::get_convolution_method(const ITensorInfo *input, const ITenso… in get_convolution_method()
161 using ConfigurationMethod = std::pair<ConvolutionConfiguration, ConvolutionMethod>; in get_convolution_method()
[all …]
/external/ComputeLibrary/src/graph/
DTypeLoader.cpp80 ConvolutionMethod Convolution_method_from_name(const std::string &name) in Convolution_method_from_name()
82 static const std::map<std::string, ConvolutionMethod> methods = in Convolution_method_from_name()
84 { "default", ConvolutionMethod::Default }, in Convolution_method_from_name()
85 { "direct", ConvolutionMethod::Direct }, in Convolution_method_from_name()
86 { "gemm", ConvolutionMethod::GEMM }, in Convolution_method_from_name()
87 { "winograd", ConvolutionMethod::Winograd }, in Convolution_method_from_name()
/external/ComputeLibrary/arm_compute/graph/nodes/
DConvolutionLayerNode.h47 ConvolutionMethod method = ConvolutionMethod::Default,
54 void set_convolution_method(ConvolutionMethod method);
62 ConvolutionMethod convolution_method() const;
117 ConvolutionMethod _method;
DFusedConvolutionBatchNormalizationNode.h48ConvolutionMethod method = ConvolutionMethod::Default,
86 void set_convolution_method(ConvolutionMethod method);
101 ConvolutionMethod convolution_method() const;
135 ConvolutionMethod _method;
DFusedDepthwiseConvolutionBatchNormalizationNode.h102 void set_convolution_method(ConvolutionMethod method);
/external/ComputeLibrary/arm_compute/graph/
DTypePrinter.h232 inline ::std::ostream &operator<<(::std::ostream &os, const ConvolutionMethod &method)
236 case ConvolutionMethod::Default:
239 case ConvolutionMethod::Direct:
242 case ConvolutionMethod::GEMM:
245 case ConvolutionMethod::Winograd:
DTypeLoader.h87 ConvolutionMethod Convolution_method_from_name(const std::string &name);
96 inline ::std::istream &operator>>(::std::istream &stream, ConvolutionMethod &target)
/external/ComputeLibrary/tests/validation/CL/
DConvolutionLayer.cpp160 … framework::dataset::make("Expected",{ ConvolutionMethod::GEMM,
161ConvolutionMethod::GEMM,
162ConvolutionMethod::GEMM,
163ConvolutionMethod::WINOGRAD,
164ConvolutionMethod::GEMM,
165ConvolutionMethod::GEMM,
166ConvolutionMethod::WINOGRAD,
167ConvolutionMethod::GEMM,
168ConvolutionMethod::GEMM,
172ConvolutionMethod is_valid = CLConvolutionLayer::get_convolution_method(&input_info.clone()->set_i…
DDilatedConvolutionLayer.cpp103 …e("Expected", { ConvolutionMethod::GEMM, ConvolutionMethod::GEMM, ConvolutionMethod::WINOGRAD, Con…
106ConvolutionMethod is_valid = CLConvolutionLayer::get_convolution_method(&input_info.clone()->set_i…
/external/ComputeLibrary/arm_compute/graph/frontend/
DTypes.h51 using graph::ConvolutionMethod;
64ConvolutionMethod convolution_method_hint = { ConvolutionMethod::Default }; …
DIStreamOperators.h82 inline IStream &operator<<(IStream &s, ConvolutionMethod convolution_method_hint)
/external/ComputeLibrary/tests/validate_examples/
Dgraph_convolution.cpp102 const std::set<arm_compute::graph::ConvolutionMethod> supported_convolution_methods in ConvolutionOptions()
104 arm_compute::graph::ConvolutionMethod::Default, in ConvolutionOptions()
105 arm_compute::graph::ConvolutionMethod::GEMM, in ConvolutionOptions()
106 arm_compute::graph::ConvolutionMethod::Winograd, in ConvolutionOptions()
107 arm_compute::graph::ConvolutionMethod::Direct in ConvolutionOptions()
117 …<arm_compute::graph::ConvolutionMethod>>("convolution_method", supported_convolution_methods, arm_… in ConvolutionOptions()
236 … EnumOption<arm_compute::graph::ConvolutionMethod> *conv_mode; /**< Convolution method */
294 if(_params.convolution_method == arm_compute::graph::ConvolutionMethod::Winograd in relative_tolerance()
/external/ComputeLibrary/tests/validation/NEON/
DDilatedConvolutionLayer.cpp96 …::dataset::make("Expected", { ConvolutionMethod::GEMM, ConvolutionMethod::GEMM, ConvolutionMethod:…
99ConvolutionMethod is_valid = NEConvolutionLayer::get_convolution_method(&input_info.clone()->set_i…
DConvolutionLayer.cpp138 …taset::make("Expected", { ConvolutionMethod::WINOGRAD, ConvolutionMethod::WINOGRAD, ConvolutionMet…
141ConvolutionMethod is_valid = NEConvolutionLayer::get_convolution_method(&input_info.clone()->set_i…
/external/ComputeLibrary/src/graph/nodes/
DFusedConvolutionBatchNormalizationNode.cpp37ConvolutionMethod method, in FusedConvolutionBatchNormalizationNode()
46 void FusedConvolutionBatchNormalizationNode::set_convolution_method(ConvolutionMethod method) in set_convolution_method()
56 ConvolutionMethod FusedConvolutionBatchNormalizationNode::convolution_method() const in convolution_method()
DConvolutionLayerNode.cpp37 ConvolutionMethod method, in ConvolutionLayerNode()
46 void ConvolutionLayerNode::set_convolution_method(ConvolutionMethod method) in set_convolution_method()
51 ConvolutionMethod ConvolutionLayerNode::convolution_method() const in convolution_method()
/external/armnn/src/backends/cl/workloads/
DClConvolution2dWorkload.hpp38 arm_compute::ConvolutionMethod GetConvolutionMethod() const;
46 arm_compute::ConvolutionMethod m_ConvolutionMethod;
/external/armnn/src/backends/neon/workloads/
DNeonConvolution2dWorkload.hpp39 arm_compute::ConvolutionMethod GetConvolutionMethod() const;
47 arm_compute::ConvolutionMethod m_ConvolutionMethod;
/external/ComputeLibrary/examples/
Dgraph_edsr.h697 arm_compute::graph::ConvolutionMethod::Default, in setup()
714 arm_compute::graph::ConvolutionMethod::Default, in setup()
745 arm_compute::graph::ConvolutionMethod::Default, in setup()
776 arm_compute::graph::ConvolutionMethod::Default, in setup()
807 arm_compute::graph::ConvolutionMethod::Default, in setup()
838 arm_compute::graph::ConvolutionMethod::Default, in setup()
869 arm_compute::graph::ConvolutionMethod::Default, in setup()
900 arm_compute::graph::ConvolutionMethod::Default, in setup()
931 arm_compute::graph::ConvolutionMethod::Default, in setup()
962 arm_compute::graph::ConvolutionMethod::Default, in setup()
[all …]
/external/ComputeLibrary/src/graph/backends/GLES/
DGCNodeValidator.cpp79 const ConvolutionMethod conv_algorithm = node.convolution_method(); in validate_convolution_layer()
83 if(conv_algorithm == ConvolutionMethod::Direct) in validate_convolution_layer()
/external/ComputeLibrary/src/graph/mutators/
DGroupedConvolutionMutator.cpp46 …PadStrideInfo conv_info, ConvolutionMethod method, ActivationLayerInfo fused_act, FastMathHint fas… in create_grouped_convolution()
143 const ConvolutionMethod conv_method = conv_node->convolution_method(); in mutate()
DNodeExecutionMethodMutator.cpp89 casted_node->set_convolution_method(ConvolutionMethod::Default); in mutate()
/external/ComputeLibrary/arm_compute/graph/backends/
DValidateHelpers.h155 const ConvolutionMethod conv_algorithm = node.convolution_method(); in validate_convolution_layer()
163 case ConvolutionMethod::Direct: in validate_convolution_layer()
167 case ConvolutionMethod::GEMM: in validate_convolution_layer()
171 case ConvolutionMethod::Winograd: in validate_convolution_layer()
175 case ConvolutionMethod::Default: in validate_convolution_layer()
/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
DNEConvolutionLayer.h150 …static ConvolutionMethod get_convolution_method(const ITensorInfo *input, const ITensorInfo *weigh…

12