Home
last modified time | relevance | path

Searched refs:per_w (Results 1 – 6 of 6) sorted by relevance

/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_uni_binary.cpp171 && conf_.bcast_type == bcast_t::per_w) in init()
179 && conf_.bcast_type == bcast_t::per_w)); in init()
192 if (conf_.bcast_type == bcast_t::per_w) { in init()
227 return bcast_t::per_w; in get_bcast_type()
1005 && (with_postops || point_broadcast || bcast_type == bcast_t::per_w in execute()
1016 else if (bcast_type == bcast_t::per_w) in execute()
H A Djit_uni_binary_kernel.cpp60 && (is_tail_kernel_ || conf_.bcast_type == bcast_t::per_w)) in get_tail_size()
72 nelems = conf_.bcast_type == bcast_t::per_w in get_tail_size()
H A Djit_primitive_conf.hpp955 per_w enumerator
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_uni_binary.cpp139 && conf_.bcast_type == bcast_t::per_w) in init()
146 && conf_.bcast_type == bcast_t::per_w)); in init()
182 return bcast_t::per_w; in get_bcast_type()
852 && (with_postops || point_broadcast || bcast_type == bcast_t::per_w in execute()
859 else if (bcast_type == bcast_t::per_w) in execute()
H A Djit_uni_binary_kernel.cpp59 && (is_tail_kernel_ || conf_.bcast_type == bcast_t::per_w)) in get_tail_size()
68 nelems = conf_.bcast_type == bcast_t::per_w in get_tail_size()
H A Djit_primitive_conf.hpp897 per_w enumerator