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 NGRAPH_RTTI_DECLARATION;
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;
47 bool evaluate_lower(
const HostTensorVector& outputs)
const override;
48 bool evaluate_upper(
const HostTensorVector& outputs)
const override;
49 bool constant_fold(OutputVector& output_values,
50 const OutputVector& inputs_values)
override;
52 virtual std::shared_ptr<Node>
53 clone_with_new_inputs(
const OutputVector& new_args)
const override;
60 NGRAPH_SUPPRESS_DEPRECATED_END
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:71
A handle for one of a node's outputs.
Definition: node_output.hpp:42
Definition: squeeze.hpp:35
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28