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

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

pub struct SubtractSpec;

impl OpSpec for SubtractSpec {
    const KIND: OneDNNGraphOpType = dnnl_graph_op_subtract;
}

impl SubtractSpec {
    pub const AUTO_BROADCAST: dnnl_graph_op_attr_t::Type =
        dnnl_graph_op_attr_t::dnnl_graph_op_attr_auto_broadcast;
}