Searched refs:lrn_desc_init (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
| H A D | lrn.cpp | 32 status_t lrn_desc_init(lrn_desc_t *lrn_desc, prop_kind_t prop_kind, in lrn_desc_init() function
83 return lrn_desc_init(lrn_desc, prop_kind, alg_kind, data_desc, nullptr, in dnnl_lrn_forward_desc_init()
90 return lrn_desc_init(lrn_desc, backward_data, alg_kind, data_desc, in dnnl_lrn_backward_desc_init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/common/ |
| H A D | lrn.cpp | 32 status_t lrn_desc_init(lrn_desc_t *lrn_desc, prop_kind_t prop_kind, in lrn_desc_init() function
83 return lrn_desc_init(lrn_desc, prop_kind, alg_kind, data_desc, nullptr, in dnnl_lrn_forward_desc_init()
90 return lrn_desc_init(lrn_desc, backward_data, alg_kind, data_desc, in dnnl_lrn_backward_desc_init()
|