24 #ifndef ARM_COMPUTE_GRAPH_PRINT_LAYER_NODE_H 25 #define ARM_COMPUTE_GRAPH_PRINT_LAYER_NODE_H 52 std::ostream &
stream()
const;
64 const std::function<ITensor *(ITensor *)>
transform()
const;
73 std::ostream &_stream;
75 const std::function<ITensor *(ITensor *)> _transform;
NodeType type() const override
Returns node's type.
std::ostream & stream() const
Stream metadata accessor.
Interface for CPU tensor.
Copyright (c) 2017-2021 Arm Limited.
const std::function< ITensor *(ITensor *)> transform() const
Transform function metadata accessor.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
PrintLayerNode(std::ostream &stream, const IOFormatInfo &format_info=IOFormatInfo(), const std::function< ITensor *(ITensor *)> transform=nullptr)
Constructor.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
const IOFormatInfo format_info() const
Formatting metadata accessor.
void accept(INodeVisitor &v) override
Accepts a node visitor.