19 #include "ngraph/axis_set.hpp"
20 #include "ngraph/op/op.hpp"
21 #include "ngraph/op/util/attr_types.hpp"
22 #include "ngraph/op/util/broadcast_base.hpp"
70 clone_with_new_inputs(
const OutputVector& new_args)
const override;
76 m_mode = broadcast_spec;
84 const HostTensorVector& inputs)
const override;
87 bool broadcast_evaluate(
const HostTensorVector& outputs,
88 const HostTensorVector& inputs)
const;
134 std::shared_ptr<Node>
135 clone_with_new_inputs(
const OutputVector& new_args)
const override;
141 m_broadcast_spec = broadcast_spec;
146 const HostTensorVector& inputs)
const override;
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: broadcast_base.hpp:31
Operation which "adds" axes to an input tensor, replicating elements from the input as needed along t...
Definition: broadcast.hpp:97
bool evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const override
Evaluates the op on input_values putting results in output_values.
Broadcast()=default
Constructs a broadcast operation.
const AutoBroadcastSpec & get_broadcast_spec() const
Definition: broadcast.hpp:138
const NodeTypeInfo & get_type_info() const override
Definition: broadcast.hpp:100
Broadcast(const Output< Node > &arg, const Output< Node > &target_shape, const AutoBroadcastSpec &broadcast_spec=AutoBroadcastSpec(AutoBroadcastType::NUMPY))
Constructs a broadcast operation.
Broadcast(const Output< Node > &arg, const Output< Node > &target_shape, const Output< Node > &axes_mapping, const AutoBroadcastSpec &broadcast_spec=AutoBroadcastSpec())
Constructs a broadcast operation.
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
Operation which "adds" axes to an input tensor, replicating elements from the input as needed along t...
Definition: broadcast.hpp:33
Broadcast(const Output< Node > &arg, const Output< Node > &target_shape, const BroadcastModeSpec &broadcast_spec=BroadcastType::NUMPY)
Constructs a broadcast operation.
Broadcast(const Output< Node > &arg, const Output< Node > &target_shape, const Output< Node > &axes_mapping, const BroadcastModeSpec &broadcast_spec=BroadcastType::EXPLICIT)
Constructs a broadcast operation.
Broadcast()=default
Constructs a broadcast operation.
bool evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const override
Evaluates the op on input_values putting results in output_values.
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
std::pair< bool, AxisSet > get_broadcast_axes() const override
const NodeTypeInfo & get_type_info() const override
Definition: broadcast.hpp:36
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Implicit broadcast specification.
Definition: attr_types.hpp:323
Implicit broadcast specification.
Definition: attr_types.hpp:379