primitive_desc(mkldnn.batch_normalization_forward.desc adesc,
mkldnn.primitive_attr aattr,
mkldnn.engine aengine) |
primitive_desc(mkldnn.convolution_forward.desc adesc,
mkldnn.primitive_attr aattr,
mkldnn.engine aengine) |
primitive_desc(mkldnn.deconvolution_forward.desc adesc,
mkldnn.primitive_attr aattr,
mkldnn.engine aengine) |
primitive_desc(mkldnn.inner_product_forward.desc adesc,
mkldnn.primitive_attr aattr,
mkldnn.engine aengine) |
primitive_desc(mkldnn.memory.primitive_desc input,
mkldnn.memory.primitive_desc output,
mkldnn.primitive_attr aattr) |