Home
last modified time | relevance | path

Searched refs:ArgMinMaxLayer (Results 1 – 23 of 23) sorted by relevance

/external/armnn/src/armnn/layers/
DArgMinMaxLayer.cpp19 ArgMinMaxLayer::ArgMinMaxLayer(const ArgMinMaxDescriptor& param, const char* name) in ArgMinMaxLayer() function in armnn::ArgMinMaxLayer
24 std::unique_ptr<IWorkload> ArgMinMaxLayer::CreateWorkload(const IWorkloadFactory& factory) const in CreateWorkload()
32 ArgMinMaxLayer* ArgMinMaxLayer::Clone(Graph& graph) const in Clone()
34 return CloneBase<ArgMinMaxLayer>(graph, m_Param, GetName()); in Clone()
37 std::vector<TensorShape> ArgMinMaxLayer::InferOutputShapes(const std::vector<TensorShape>& inputSha… in InferOutputShapes()
74 void ArgMinMaxLayer::ValidateTensorShapesFromInputs() in ValidateTensorShapesFromInputs()
89 void ArgMinMaxLayer::ExecuteStrategy(IStrategy& strategy) const in ExecuteStrategy()
DArgMinMaxLayer.hpp14 class ArgMinMaxLayer : public LayerWithParameters<ArgMinMaxDescriptor> class
25 ArgMinMaxLayer* Clone(Graph& graph) const override;
43 ArgMinMaxLayer(const ArgMinMaxDescriptor& param, const char* name);
46 ~ArgMinMaxLayer() = default;
/external/armnn/src/armnnDeserializer/test/
DDeserializeArgMinMax.cpp48 layer_type: "ArgMinMaxLayer", in ArgMinMaxFixture()
52 layerName: "ArgMinMaxLayer", in ArgMinMaxFixture()
/external/ComputeLibrary/arm_compute/graph/nodes/
DArgMinMaxLayerNode.h71 static constexpr NodeType node_type = NodeType::ArgMinMaxLayer;
/external/ComputeLibrary/src/graph/backends/NEON/
DNENodeValidator.cpp67 case NodeType::ArgMinMaxLayer: in validate()
DNEFunctionFactory.cpp135 case NodeType::ArgMinMaxLayer: in create()
/external/ComputeLibrary/src/graph/backends/CL/
DCLNodeValidator.cpp68 case NodeType::ArgMinMaxLayer: in validate()
DCLFunctionsFactory.cpp243 case NodeType::ArgMinMaxLayer: in create()
/external/ComputeLibrary/arm_compute/graph/
DTypePrinter.h68 case NodeType::ArgMinMaxLayer:
DTypes.h202 ArgMinMaxLayer, enumerator
/external/armnn/src/armnnSerializer/
DArmnnSchema.fbs225 table ArgMinMaxLayer {
1095 ArgMinMaxLayer,
/external/ComputeLibrary/arm_compute/graph/backends/
DValidateHelpers.h63 template <typename ArgMinMaxLayer>
75 return ArgMinMaxLayer::validate(input, node.axis(), output, node.reduction_operation()); in validate_arg_min_max_layer()
/external/ComputeLibrary/arm_compute/graph/frontend/
DLayers.h149 class ArgMinMaxLayer final : public ILayer
159 ArgMinMaxLayer(ReductionOperation op,
/external/armnn/src/armnn/test/
DShapeInferenceTests.cpp193 CreateGraphAndRunTest<ArgMinMaxLayer>({{ 1, 3, 2, 4 }}, {{ 1, 2, 4 }}, descriptor, "argMinMax");
DInferOutputTests.hpp24 auto argMinMaxLayer = graph.AddLayer<armnn::ArgMinMaxLayer>(descriptor, "argMinMax"); in ArgMinMaxInferOutputShapeImpl()
/external/armnn/
DAndroid.bp126 "src/armnn/layers/ArgMinMaxLayer.cpp",
DCMakeLists.txt190 src/armnn/layers/ArgMinMaxLayer.hpp
191 src/armnn/layers/ArgMinMaxLayer.cpp
DLICENSE.spdx7298 FileName: ./src/armnn/layers/ArgMinMaxLayer.cpp
7305 FileName: ./src/armnn/layers/ArgMinMaxLayer.hpp
/external/armnn/generated/
DArmnnSchema_generated.h62 struct ArgMinMaxLayer;
1744 template<> struct LayerTraits<armnnSerializer::ArgMinMaxLayer> {
2768 struct ArgMinMaxLayer FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
2791 typedef ArgMinMaxLayer Table;
2795 fbb_.AddOffset(ArgMinMaxLayer::VT_BASE, base);
2798 fbb_.AddOffset(ArgMinMaxLayer::VT_DESCRIPTOR, descriptor);
2804 flatbuffers::Offset<ArgMinMaxLayer> Finish() {
2806 auto o = flatbuffers::Offset<ArgMinMaxLayer>(end);
2811 inline flatbuffers::Offset<ArgMinMaxLayer> CreateArgMinMaxLayer(
10727 const armnnSerializer::ArgMinMaxLayer *layer_as_ArgMinMaxLayer() const {
[all …]
/external/armnn/src/backends/backendsCommon/
DWorkloadFactory.cpp126 auto cLayer = PolymorphicDowncast<const ArgMinMaxLayer*>(&layer); in IsLayerConfigurationSupported()
/external/armnn/src/armnn/
DNetwork.cpp2186 return m_Graph->AddLayer<ArgMinMaxLayer>(argMinMaxDescriptor, name); in AddArgMinMaxLayer()
/external/armnn/docs/
D02_operator_list.dox215 <td rowspan="3">ArgMinMaxLayer
/external/ComputeLibrary/docs/user_guide/
Doperator_list.dox112 <td rowspan="2">ArgMinMaxLayer