21 #include <type_traits>
24 #include "ngraph/op/op.hpp"
42 uint64_t get_reduction_axis()
const;
43 void set_reduction_axis(uint64_t value);
45 void set_index_element_type(
const element::Type& index_element_type);
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: element_type.hpp:61
Root of all actual ops.
Definition: op.hpp:29
Definition: index_reduction.hpp:33
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28