onednnl 0.0.1

high-level bindings to oneDNN Deep Learning library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use {
    crate::graph::{
        op::{OneDNNGraphOp, OneDNNGraphOpType},
        spec::OpSpec,
    },
    onednnl_sys::dnnl_graph_op_attr_t,
};

pub struct AddSpec;

impl OpSpec for AddSpec {
    const KIND: OneDNNGraphOpType = OneDNNGraphOp::ADD;
}

impl AddSpec {
    /// Possible values of "none" and "numpy"
    pub const AUTO_BROADCAST: dnnl_graph_op_attr_t::Type =
        dnnl_graph_op_attr_t::dnnl_graph_op_attr_auto_broadcast;
}