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

use onednnl_sys::dnnl_graph_op_attr_t;

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

pub struct SoftmaxSpec;

impl OpSpec for SoftmaxSpec {
    const KIND: OneDNNGraphOpType = OneDNNGraphOp::SOFTMAX;
}

impl SoftmaxSpec {
    pub const AXIS: dnnl_graph_op_attr_t::Type = dnnl_graph_op_attr_t::dnnl_graph_op_attr_axis;
}