Home
last modified time | relevance | path

Searched refs:eltwise_beta (Results 1 – 25 of 47) sorted by relevance

12

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/gemm/
H A Dxe_lp_gemm.cpp40 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 Dgen9_gemm_x8x8s32.cpp45 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 Dgen9_gemm.cpp129 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 Dref_gemm.cpp70 const float eltwise_beta = pd()->attr_info.eltwise_beta; in execute() local
100 arg_list.set(21, eltwise_beta); in execute()
H A Dgen9_gemm_x8x8s32.hpp208 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 Dgen9_gemm_nocopy_scale_x8x8s32.cl25 val, eltwise_alpha, eltwise_beta, eltwise_scale); \
34 float eltwise_alpha, float eltwise_beta, float eltwise_scale) {
H A Dxe_lp_gemm_nocopy_scale_x8x8s32.cl25 val, eltwise_alpha, eltwise_beta, eltwise_scale); \
34 float eltwise_alpha, float eltwise_beta, float eltwise_scale) {
H A Dxe_lp_gemm.hpp141 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 Dref_gemm.cl38 float eltwise_alpha, float eltwise_beta, float eltwise_scale,
82 temp = fwd_eltwise(temp, eltwise_alpha, eltwise_beta, eltwise_scale);
H A Dgen9_gemm.hpp220 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 Dgen9_gemm_nocopy_superkernel_f32.cl111 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 Dgen9_gemm_x8x8s32.cpp45 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 Dxe_lp_gemm.cpp40 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 Dgen9_gemm.cpp129 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 Dref_gemm.cpp70 const float eltwise_beta = pd()->attr_info.eltwise_beta; in execute() local
100 arg_list.set(21, eltwise_beta); in execute()
H A Dgen9_gemm_x8x8s32.hpp214 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 Dxe_lp_gemm_nocopy_scale_x8x8s32.cl25 val, eltwise_alpha, eltwise_beta, eltwise_scale); \
34 float eltwise_alpha, float eltwise_beta, float eltwise_scale) {
H A Dgen9_gemm_nocopy_scale_x8x8s32.cl25 val, eltwise_alpha, eltwise_beta, eltwise_scale); \
34 float eltwise_alpha, float eltwise_beta, float eltwise_scale) {
H A Dxe_lp_gemm.hpp143 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 Dref_gemm.cl38 float eltwise_alpha, float eltwise_beta, float eltwise_scale,
82 temp = fwd_eltwise(temp, eltwise_alpha, eltwise_beta, eltwise_scale);
H A Dgen9_gemm.hpp236 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 Dgen_gemm.hpp151 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 Dgen_gemm_kernel.hpp88 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 Dtest_convolution_eltwise_forward_common.hpp178 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 Dtest_convolution_eltwise_forward_common.hpp178 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()

12