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_multiply},
};

pub struct MultiplySpec;

impl OpSpec for MultiplySpec {
    const KIND: OneDNNGraphOpType = dnnl_graph_op_multiply;
}

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