/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/gemm/ |
H A D | xe_lp_gemm.cpp | 40 float eltwise_beta, float eltwise_scale, bool aligned) const { in launch_x8x8s32() argument 77 arg_list.set(22, eltwise_beta); in launch_x8x8s32() 110 bool apply_eltwise, float eltwise_alpha, float eltwise_beta, in launch_scale_x8x8s32() argument 131 arg_list.set(14, eltwise_beta); in launch_scale_x8x8s32() 197 auto eltwise_beta = pd()->eltwise_beta(); in execute_standard() local 270 eltwise_alpha, eltwise_beta, eltwise_scale, in execute_standard() 286 eltwise_alpha, eltwise_beta, eltwise_scale, in execute_standard() 298 (int)alpha_is_zero, true, eltwise_alpha, eltwise_beta, in execute_standard()
|
H A D | gen9_gemm_x8x8s32.cpp | 45 float eltwise_beta, float eltwise_scale) const { in launch_x8x8s32() argument 79 arg_list.set(22, eltwise_beta); in launch_x8x8s32() 109 float eltwise_beta, float eltwise_scale) const { in launch_scale_x8x8s32() argument 126 arg_list.set(14, eltwise_beta); in launch_scale_x8x8s32() 195 auto eltwise_beta = pd()->attr_info.eltwise_beta; in execute_standard() local 266 (int)apply_eltwise, eltwise_alpha, eltwise_beta, in execute_standard() 275 eltwise_beta, eltwise_scale); in execute_standard() 286 (int)alpha_is_zero, true, eltwise_alpha, eltwise_beta, in execute_standard()
|
H A D | gen9_gemm.cpp | 129 int64_t ldc, int last_k_block, float eltwise_alpha, float eltwise_beta, in launch_compute() argument 144 arg_list.set(11, eltwise_beta); in launch_compute() 170 int last_k_block, float eltwise_alpha, float eltwise_beta, in launch_nocopy() argument 192 arg_list.set(16, eltwise_beta); in launch_nocopy() 242 float eltwise_alpha, float eltwise_beta, float eltwise_scale) const { in launch_nocopy_superkernel() argument 263 arg_list.set(18, eltwise_beta); in launch_nocopy_superkernel() 311 auto eltwise_beta = pd()->eltwise_beta(); in execute_standard() local 398 (int)last_k_block, eltwise_alpha, eltwise_beta, in execute_standard() 405 eltwise_alpha, eltwise_beta, eltwise_scale, in execute_standard() 440 auto eltwise_beta = pd()->eltwise_beta(); in execute_superkernel() local [all …]
|
H A D | ref_gemm.cpp | 70 const float eltwise_beta = pd()->attr_info.eltwise_beta; in execute() local 100 arg_list.set(21, eltwise_beta); in execute()
|
H A D | gen9_gemm_x8x8s32.hpp | 208 float eltwise_alpha, float eltwise_beta, float eltwise_scale) const; 215 bool apply_eltwise, float eltwise_alpha, float eltwise_beta, 223 bool apply_eltwise, float eltwise_alpha, float eltwise_beta) const;
|
H A D | gen9_gemm_nocopy_scale_x8x8s32.cl | 25 val, eltwise_alpha, eltwise_beta, eltwise_scale); \ 34 float eltwise_alpha, float eltwise_beta, float eltwise_scale) {
|
H A D | xe_lp_gemm_nocopy_scale_x8x8s32.cl | 25 val, eltwise_alpha, eltwise_beta, eltwise_scale); \ 34 float eltwise_alpha, float eltwise_beta, float eltwise_scale) {
|
H A D | xe_lp_gemm.hpp | 141 float eltwise_beta() const { in eltwise_beta() function 261 float eltwise_alpha, float eltwise_beta, float eltwise_scale, 269 float eltwise_alpha, float eltwise_beta, float eltwise_scale) const;
|
H A D | ref_gemm.cl | 38 float eltwise_alpha, float eltwise_beta, float eltwise_scale, 82 temp = fwd_eltwise(temp, eltwise_alpha, eltwise_beta, eltwise_scale);
|
H A D | gen9_gemm.hpp | 220 float eltwise_beta() const { in eltwise_beta() function 511 int last_k_block, float eltwise_alpha, float eltwise_beta, 520 float eltwise_beta, float eltwise_scale, 530 float eltwise_beta, float eltwise_scale) const;
|
H A D | gen9_gemm_nocopy_superkernel_f32.cl | 111 val, eltwise_alpha, eltwise_beta, eltwise_scale); \ 206 float eltwise_beta, float eltwise_scale) { 314 float eltwise_beta, float eltwise_scale) {
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/gemm/ |
H A D | gen9_gemm_x8x8s32.cpp | 45 float eltwise_beta, float eltwise_scale) const { in launch_x8x8s32() argument 79 arg_list.set(22, eltwise_beta); in launch_x8x8s32() 109 float eltwise_beta, float eltwise_scale) const { in launch_scale_x8x8s32() argument 126 arg_list.set(14, eltwise_beta); in launch_scale_x8x8s32() 195 auto eltwise_beta = pd()->attr_info.eltwise_beta; in execute_standard() local 266 (int)apply_eltwise, eltwise_alpha, eltwise_beta, in execute_standard() 275 eltwise_beta, eltwise_scale); in execute_standard() 286 (int)alpha_is_zero, true, eltwise_alpha, eltwise_beta, in execute_standard()
|
H A D | xe_lp_gemm.cpp | 40 float eltwise_beta, float eltwise_scale, bool aligned) const { in launch_x8x8s32() argument 77 arg_list.set(22, eltwise_beta); in launch_x8x8s32() 110 bool apply_eltwise, float eltwise_alpha, float eltwise_beta, in launch_scale_x8x8s32() argument 131 arg_list.set(14, eltwise_beta); in launch_scale_x8x8s32() 196 auto eltwise_beta = pd()->eltwise_beta(); in execute_standard() local 267 eltwise_alpha, eltwise_beta, eltwise_scale, in execute_standard() 283 eltwise_alpha, eltwise_beta, eltwise_scale, in execute_standard() 295 (int)alpha_is_zero, true, eltwise_alpha, eltwise_beta, in execute_standard()
|
H A D | gen9_gemm.cpp | 129 int64_t ldc, int last_k_block, float eltwise_alpha, float eltwise_beta, in launch_compute() argument 144 arg_list.set(11, eltwise_beta); in launch_compute() 170 int last_k_block, float eltwise_alpha, float eltwise_beta, in launch_nocopy() argument 192 arg_list.set(16, eltwise_beta); in launch_nocopy() 242 float eltwise_alpha, float eltwise_beta, float eltwise_scale) const { in launch_nocopy_superkernel() argument 263 arg_list.set(18, eltwise_beta); in launch_nocopy_superkernel() 311 auto eltwise_beta = pd()->eltwise_beta(); in execute_standard() local 398 (int)last_k_block, eltwise_alpha, eltwise_beta, in execute_standard() 405 eltwise_alpha, eltwise_beta, eltwise_scale, in execute_standard() 440 auto eltwise_beta = pd()->eltwise_beta(); in execute_superkernel() local [all …]
|
H A D | ref_gemm.cpp | 70 const float eltwise_beta = pd()->attr_info.eltwise_beta; in execute() local 100 arg_list.set(21, eltwise_beta); in execute()
|
H A D | gen9_gemm_x8x8s32.hpp | 214 float eltwise_alpha, float eltwise_beta, float eltwise_scale) const; 221 bool apply_eltwise, float eltwise_alpha, float eltwise_beta, 229 bool apply_eltwise, float eltwise_alpha, float eltwise_beta) const;
|
H A D | xe_lp_gemm_nocopy_scale_x8x8s32.cl | 25 val, eltwise_alpha, eltwise_beta, eltwise_scale); \ 34 float eltwise_alpha, float eltwise_beta, float eltwise_scale) {
|
H A D | gen9_gemm_nocopy_scale_x8x8s32.cl | 25 val, eltwise_alpha, eltwise_beta, eltwise_scale); \ 34 float eltwise_alpha, float eltwise_beta, float eltwise_scale) {
|
H A D | xe_lp_gemm.hpp | 143 float eltwise_beta() const { in eltwise_beta() function 265 float eltwise_alpha, float eltwise_beta, float eltwise_scale, 273 float eltwise_alpha, float eltwise_beta, float eltwise_scale) const;
|
H A D | ref_gemm.cl | 38 float eltwise_alpha, float eltwise_beta, float eltwise_scale, 82 temp = fwd_eltwise(temp, eltwise_alpha, eltwise_beta, eltwise_scale);
|
H A D | gen9_gemm.hpp | 236 float eltwise_beta() const { in eltwise_beta() function 530 int last_k_block, float eltwise_alpha, float eltwise_beta, 539 float eltwise_beta, float eltwise_scale, 549 float eltwise_beta, float eltwise_scale) const;
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/jit/gemm/ |
H A D | gen_gemm.hpp | 151 float eltwise_beta() const { return attr_info()->eltwise_beta; } in eltwise_beta() function 204 auto eltwise_beta = pd()->eltwise_beta(); in init_nocopy() local 215 eltwise_alpha, eltwise_beta, eltwise_scale, a_type, b_type, in init_nocopy()
|
H A D | gen_gemm_kernel.hpp | 88 float eltwise_alpha, float eltwise_beta, float eltwise_scale, in init() 119 problem_.eltwiseBeta = eltwise_beta; in init()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
H A D | test_convolution_eltwise_forward_common.hpp | 178 float eltwise_beta = p.eltwise_beta; in Test() local 240 ops.append_eltwise(1.0, p.alg, p.eltwise_alpha, p.eltwise_beta); in Test() 267 p.alg, eltwise_alpha, eltwise_beta); in Test()
|
/dports/math/onednn/oneDNN-2.5.1/tests/gtests/ |
H A D | test_convolution_eltwise_forward_common.hpp | 178 float eltwise_beta = p.eltwise_beta; in Test() local 240 ops.append_eltwise(1.0, p.alg, p.eltwise_alpha, p.eltwise_beta); in Test() 267 p.alg, eltwise_alpha, eltwise_beta); in Test()
|