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

use {
    crate::graph::spec::{OpSpec, RequiredAttrs},
    onednnl_sys::{dnnl_graph_op_attr_t::dnnl_graph_op_attr_mode, dnnl_graph_op_kind_t},
};

pub struct GeluSpec;

impl OpSpec for GeluSpec {
    const KIND: dnnl_graph_op_kind_t::Type = dnnl_graph_op_kind_t::dnnl_graph_op_gelu;
}

pub struct GeluAttrs {
    pub mode: String,
}

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