/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/lrn/ |
H A D | jit_avx512_common_lrn.cpp | 53 && (desc()->lrn_beta == 0.75 || desc()->lrn_beta == 1.0) in init() 102 && (desc()->lrn_beta == 0.75 || desc()->lrn_beta == 1.0) in init()
|
H A D | lrn_avx512_nhwc_executor.hpp | 38 pd->desc()->lrn_beta, pd->desc()->lrn_k, in lrn_avx512_nhwc_executor_fwd_t() 92 pd->desc()->lrn_beta, pd->desc()->local_size)} in lrn_avx512_nhwc_executor_bwd_t()
|
H A D | jit_uni_lrn.cpp | 179 && data_d.dims()[1] >= 2 * VECTOR_LENGTH && desc()->lrn_beta == 0.75 in init() 239 const float &B = pd()->desc()->lrn_beta; in init() 345 && data_d.dims()[1] >= 2 * VECTOR_LENGTH && desc()->lrn_beta == 0.75 in init()
|
H A D | lrn_avx512_blocked_executor.hpp | 45 const float beta = pd->desc()->lrn_beta; in lrn_avx512_blocked_executor_fwd_t() 185 const float beta = pd->desc()->lrn_beta; in lrn_avx512_blocked_executor_bwd_t()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/lrn/ |
H A D | jit_avx512_common_lrn.cpp | 53 && (desc()->lrn_beta == 0.75 || desc()->lrn_beta == 1.0) in init() 102 && (desc()->lrn_beta == 0.75 || desc()->lrn_beta == 1.0) in init()
|
H A D | lrn_avx512_nhwc_executor.hpp | 38 pd->desc()->lrn_beta, pd->desc()->lrn_k, in lrn_avx512_nhwc_executor_fwd_t() 92 pd->desc()->lrn_beta, pd->desc()->local_size)} in lrn_avx512_nhwc_executor_bwd_t()
|
H A D | jit_uni_lrn.cpp | 179 && data_d.dims()[1] >= 2 * VECTOR_LENGTH && desc()->lrn_beta == 0.75 in init() 239 const float &B = pd()->desc()->lrn_beta; in init() 345 && data_d.dims()[1] >= 2 * VECTOR_LENGTH && desc()->lrn_beta == 0.75 in init()
|
H A D | lrn_avx512_blocked_executor.hpp | 45 const float beta = pd->desc()->lrn_beta; in lrn_avx512_blocked_executor_fwd_t() 185 const float beta = pd->desc()->lrn_beta; in lrn_avx512_blocked_executor_bwd_t()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/nvidia/ |
H A D | cudnn_lrn_impl.hpp | 58 double lrn_beta; member 82 lrn_beta = lrn_desc->lrn_beta; in init_common() 105 lrn_alpha, lrn_beta, lrn_K)); in create_and_set_lrn_descriptor()
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/nvidia/ |
H A D | cudnn_lrn_impl.hpp | 58 double lrn_beta; member 82 lrn_beta = lrn_desc->lrn_beta; in init_common() 105 lrn_alpha, lrn_beta, lrn_K)); in create_and_set_lrn_descriptor()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
H A D | lrn.cpp | 64 ld.lrn_beta = beta; in lrn_desc_init()
|
H A D | type_helpers.hpp | 448 && COMPARE_FLOAT_DESC_MEMBERS(lrn_beta) in operator ==()
|
H A D | primitive_hashing.cpp | 454 seed = hash_combine(seed, desc.lrn_beta); in get_desc_hash()
|
H A D | verbose.cpp | 632 ss << "ls" << pd->desc()->local_size << "beta" << pd->desc()->lrn_beta; in init_info_lrn()
|
/dports/math/onednn/oneDNN-2.5.1/src/common/ |
H A D | lrn.cpp | 64 ld.lrn_beta = beta; in lrn_desc_init()
|
H A D | type_helpers.hpp | 427 && COMPARE_FLOAT_DESC_MEMBERS(lrn_beta) in operator ==()
|
H A D | primitive_hashing.cpp | 461 seed = hash_combine(seed, desc.lrn_beta); in get_desc_hash()
|
H A D | verbose.cpp | 670 ss << "ls" << pd->desc()->local_size << "beta" << pd->desc()->lrn_beta; in init_info_lrn()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
H A D | ref_lrn.hpp | 137 kernel_ctx.define_float("LRN_BETA", desc->lrn_beta); in init() 260 kernel_ctx.define_float("LRN_BETA", desc->lrn_beta); in init()
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
H A D | ref_lrn.hpp | 137 kernel_ctx.define_float("LRN_BETA", desc->lrn_beta); in init() 260 kernel_ctx.define_float("LRN_BETA", desc->lrn_beta); in init()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
H A D | ref_lrn.cpp | 93 const acc_data_t beta = static_cast<acc_data_t>(pd()->desc()->lrn_beta); in execute_forward() 220 const acc_data_t beta = static_cast<acc_data_t>(pd()->desc()->lrn_beta); in execute_backward()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
H A D | ref_lrn.cpp | 93 const acc_data_t beta = static_cast<acc_data_t>(pd()->desc()->lrn_beta); in execute_forward() 220 const acc_data_t beta = static_cast<acc_data_t>(pd()->desc()->lrn_beta); in execute_backward()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/include/oneapi/dnnl/ |
H A D | dnnl_types.h | 1931 float lrn_beta; member
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/oneapi/dnnl/ |
H A D | dnnl_types.h | 1931 float lrn_beta; member
|
/dports/math/onednn/oneDNN-2.5.1/include/oneapi/dnnl/ |
H A D | dnnl_types.h | 2142 float lrn_beta; member
|