onnx_helpers/nodes/ops/
size.rs

1//! Size operation.
2
3use crate::{builder, node_to_inner, nodes::Node};
4
5/// Size node.
6pub struct Size {
7    inner: Node,
8}
9
10impl Size {
11    /// Creates new Size operation.
12    #[inline(always)]
13    pub fn new<T: Into<String>>(input: T) -> Self {
14        Size {
15            inner: builder::Node::new("Size").input(input).build(),
16        }
17    }
18}
19
20node_to_inner!(Size);