21 #include "ngraph/axis_vector.hpp"
22 #include "ngraph/node.hpp"
23 #include "ngraph/op/op.hpp"
24 #include "ngraph/op/util/fused_op.hpp"
26 NGRAPH_SUPPRESS_DEPRECATED_START
34 class NGRAPH_API
Squeeze :
public ngraph::op::util::FusedOp
37 static constexpr NodeTypeInfo type_info{
"Squeeze", 0};
38 const NodeTypeInfo& get_type_info()
const override {
return type_info; }
40 Squeeze(
const Output<Node>& data,
const Output<Node>& axes);
43 virtual OutputVector decompose_op()
const override;
44 virtual void pre_validate_and_infer_types()
override;
45 bool evaluate(
const HostTensorVector& outputs,
46 const HostTensorVector& inputs)
const override;
48 virtual std::shared_ptr<Node>
49 clone_with_new_inputs(
const OutputVector& new_args)
const override;
56 NGRAPH_SUPPRESS_DEPRECATED_END