Searched refs:jit_relu_ (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_uni_tbb_batch_normalization.cpp | 774 jit_bnorm_process_relu_t<isa> jit_relu_; member
852 jit_relu_.fwd_process_relu(v_); in compute_bnorm()
963 , jit_relu_(bdesc, this, reg_off_dat_, reg_tmp_, reg_ptr_ws_, vzero_, in jit_bnorm_fwd_t()
987 jit_relu_.fwd_prepare_relu(); in generate()
1059 jit_bnorm_process_relu_t<isa> jit_relu_; member
1130 jit_relu_.bwd_process_relu(v_); in compute_bnorm()
1251 , jit_relu_(bdesc, this, reg_off_dat_, reg_tmp_, reg_ptr_ws_, vzero_, in jit_bnorm_bwd_t()
1274 jit_relu_.bwd_prepare_relu(); in generate()
1354 jit_bnorm_process_relu_t<isa> jit_relu_; member
1481 jit_relu_.bwd_process_relu( in compute_diff_beta_and_diff_gamma()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_uni_tbb_batch_normalization.cpp | 772 jit_bnorm_process_relu_t<isa> jit_relu_; member
850 jit_relu_.fwd_process_relu(v_); in compute_bnorm()
961 , jit_relu_(bdesc, this, reg_off_dat_, reg_tmp_, reg_ptr_ws_, vzero_, in jit_bnorm_fwd_t()
985 jit_relu_.fwd_prepare_relu(); in generate()
1057 jit_bnorm_process_relu_t<isa> jit_relu_; member
1128 jit_relu_.bwd_process_relu(v_); in compute_bnorm()
1249 , jit_relu_(bdesc, this, reg_off_dat_, reg_tmp_, reg_ptr_ws_, vzero_, in jit_bnorm_bwd_t()
1272 jit_relu_.bwd_prepare_relu(); in generate()
1352 jit_bnorm_process_relu_t<isa> jit_relu_; member
1479 jit_relu_.bwd_process_relu( in compute_diff_beta_and_diff_gamma()
[all …]
|