onnx_helpers/nodes/ops/
reduce_min.rs

1//! Reduce mean operation.
2
3use onnx_pb::Axes;
4
5use crate::{builder, node_to_inner, nodes::Node};
6
7/// Reduce mean node.
8pub struct ReduceMin {
9    inner: Node,
10}
11
12impl ReduceMin {
13    /// Creates new reduce mean operation.
14    #[inline(always)]
15    pub fn new<S: Into<String>, A: Into<Axes>>(input: S, axes: A, keepdims: bool) -> Self {
16        ReduceMin {
17            inner: builder::Node::new("ReduceMin")
18                .input(input)
19                .attribute("axes", axes.into())
20                .attribute("keepdims", keepdims)
21                .build(),
22        }
23    }
24}
25
26node_to_inner!(ReduceMin);