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
20
21
22
23

use onednnl_sys::dnnl_graph_op_attr_t::dnnl_graph_op_attr_axis;

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

pub struct ConcatSpec;

impl OpSpec for ConcatSpec {
    const KIND: OneDNNGraphOpType = OneDNNGraphOp::CONCAT;
}

#[derive(Debug, Clone, Copy)]
pub struct ConcatAttrs {
    pub axis: i64,
}

impl From<ConcatAttrs> for RequiredAttrs {
    fn from(attrs: ConcatAttrs) -> Self {
        RequiredAttrs::Some(vec![(dnnl_graph_op_attr_axis, vec![attrs.axis].into())])
    }
}