ngraph::onnx_import::utils::ArgMinMaxFactory Class Reference

Factory class which generates sub-graphs for ONNX ArgMin, ArgMax ops. More...

#include <arg_min_max_factory.hpp>

Public Member Functions
	ArgMinMaxFactory (const Node &node)
std::shared_ptr< ngraph::Node >	make_arg_max () const
	Creates ArgMax ONNX operation. More...
std::shared_ptr< ngraph::Node >	make_arg_min () const
	Creates ArgMin ONNX operation. More...

Detailed Description

Factory class which generates sub-graphs for ONNX ArgMin, ArgMax ops.

Member Function Documentation

make_arg_max()

std::shared_ptr<ngraph::Node> ngraph::onnx_import::utils::ArgMinMaxFactory::make_arg_max

(

)

const

Creates ArgMax ONNX operation.

Returns

Sub-graph representing ArgMax op.

make_arg_min()

std::shared_ptr<ngraph::Node> ngraph::onnx_import::utils::ArgMinMaxFactory::make_arg_min

(

)

const

Creates ArgMin ONNX operation.

Returns

Sub-graph representing ArgMin op.

---

The documentation for this class was generated from the following file:

• frontend/onnx_import/include/onnx_import/utils/arg_min_max_factory.hpp