19 #include "ngraph/op/op.hpp"
29 class NGRAPH_DEPRECATED(
30 "This operation is deprecated and will be removed soon. "
31 "Use v1::Softmax instead of it.") NGRAPH_API Softmax :
public Op
33 NGRAPH_SUPPRESS_DEPRECATED_START
35 static constexpr NodeTypeInfo type_info{
"Softmax", 0};
36 const NodeTypeInfo& get_type_info()
const override {
return type_info; }
46 Softmax(
const Output<Node>& arg,
const AxisSet& axes);
56 Softmax(
const Output<Node>& arg,
const Output<Node>& axes);
58 void validate_and_infer_types()
override;
60 virtual std::shared_ptr<Node>
61 clone_with_new_inputs(
const OutputVector& new_args)
const override;
63 bool are_axes_constant()
const;
64 const AxisSet get_axes()
const;
65 void set_axes(
const AxisSet& axes);
67 bool evaluate(
const HostTensorVector& outputs,
68 const HostTensorVector& inputs)
const override;
69 NGRAPH_SUPPRESS_DEPRECATED_END
78 static constexpr NodeTypeInfo type_info{
"Softmax", 1};
79 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
92 Softmax(
const Output<Node>& arg,
const size_t axis);
98 virtual std::shared_ptr<Node>
99 clone_with_new_inputs(
const OutputVector& new_args)
const override;
101 size_t get_axis()
const {
return m_axis; }
102 void set_axis(
const size_t axis) { m_axis = axis; }
103 bool evaluate(
const HostTensorVector& outputs,
104 const HostTensorVector& inputs)
const override;
112 NGRAPH_SUPPRESS_DEPRECATED_START
114 NGRAPH_SUPPRESS_DEPRECATED_END