Home
last modified time | relevance | path

Searched refs:rcpstddev (Results 1 – 23 of 23) sorted by relevance

/dports/math/libxsmm/libxsmm-1.16.3/src/
H A Dlibxsmm_dnn_fusedbatchnorm_forward.c350 handle->expvalue == 0 || handle->rcpstddev == 0 || handle->variance == 0 ) { in libxsmm_dnn_fusedbatchnorm_st_fwd_custom()
515 …if ( handles[l_count]->expvalue == 0 || handles[l_count]->rcpstddev == 0 || handles[l_count]->va… in libxsmm_dnn_fusedbatchnorm_reduce_stats_st_fwd_custom()
546 LIBXSMM_VLA_DECL(2, float, brstd0, (float*)handles[0]->rcpstddev->data, nFmBlock); in libxsmm_dnn_fusedbatchnorm_reduce_stats_st_fwd_custom()
595 LIBXSMM_VLA_DECL(2, float, brstdr, (float*)handles[l_count]->rcpstddev->data, nFmBlock); in libxsmm_dnn_fusedbatchnorm_reduce_stats_st_fwd_custom()
H A Dlibxsmm_dnn_fusedbatchnorm.c430 handle->rcpstddev = (libxsmm_dnn_tensor*)tensor; in libxsmm_dnn_fusedbatchnorm_bind_tensor()
493 return_tensor = handle->rcpstddev; in libxsmm_dnn_fusedbatchnorm_get_tensor()
548 handle->rcpstddev = 0; in libxsmm_dnn_fusedbatchnorm_release_tensor()
H A Dlibxsmm_dnn_fusedgroupnorm.c450 handle->rcpstddev = (libxsmm_dnn_tensor*)tensor; in libxsmm_dnn_fusedgroupnorm_bind_tensor()
513 return_tensor = handle->rcpstddev; in libxsmm_dnn_fusedgroupnorm_get_tensor()
568 handle->rcpstddev = 0; in libxsmm_dnn_fusedgroupnorm_release_tensor()
H A Dlibxsmm_main.h525 libxsmm_dnn_tensor* rcpstddev; /* reciprocal of standard derivation */ member
581 libxsmm_dnn_tensor* rcpstddev; /* reciprocal of standard derivation */ member
H A Dlibxsmm_dnn_fusedgroupnorm_forward.c344 handle->expvalue == 0 || handle->rcpstddev == 0 || handle->variance == 0 ) { in libxsmm_dnn_fusedgroupnorm_st_fwd_custom()
H A Dlibxsmm_dnn_fusedgroupnorm_backward.c337 handle->expvalue == 0 || handle->rcpstddev == 0 ) { in libxsmm_dnn_fusedgroupnorm_st_bwd_custom()
H A Dlibxsmm_dnn_fusedbatchnorm_backward.c343 handle->expvalue == 0 || handle->rcpstddev == 0 ) { in libxsmm_dnn_fusedbatchnorm_st_bwd_custom()
/dports/math/libxsmm/libxsmm-1.16.3/src/template/
H A Dlibxsmm_dnn_fusedgroupnorm_st_fwd_custom_generic.tpl.c71 LIBXSMM_VLA_DECL(2, element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->da…
H A Dlibxsmm_dnn_fusedgroupnorm_st_fwd_custom_f32_bf16_c16_avx512.tpl.c87 LIBXSMM_VLA_DECL(2, element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->da…
H A Dlibxsmm_dnn_fusedgroupnorm_st_bwd_custom_f32_bf16_c16_avx512.tpl.c91 LIBXSMM_VLA_DECL(2, const element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->…
H A Dlibxsmm_dnn_fusedbatchnorm_st_fwd_custom_f32_bf16_c16_avx512.tpl.c87 LIBXSMM_VLA_DECL(2, element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->da…
H A Dlibxsmm_dnn_fusedbatchnorm_st_fwd_custom_generic.tpl.c73 LIBXSMM_VLA_DECL(2, element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->da…
H A Dlibxsmm_dnn_fusedgroupnorm_st_fwd_custom_f32_bf16_c32_avx512.tpl.c87 LIBXSMM_VLA_DECL(2, element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->da…
H A Dlibxsmm_dnn_fusedbatchnorm_st_bwd_custom_f32_bf16_c16_avx512.tpl.c91 LIBXSMM_VLA_DECL(2, const element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->…
H A Dlibxsmm_dnn_fusedbatchnorm_st_fwd_custom_f32_bf16_c32_avx512.tpl.c87 LIBXSMM_VLA_DECL(2, element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->da…
H A Dlibxsmm_dnn_fusedbatchnorm_st_bwd_custom_generic.tpl.c77 LIBXSMM_VLA_DECL(2, const element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->…
H A Dlibxsmm_dnn_fusedgroupnorm_st_bwd_custom_f32_bf16_c32_avx512.tpl.c91 LIBXSMM_VLA_DECL(2, const element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->…
H A Dlibxsmm_dnn_fusedgroupnorm_st_bwd_custom_generic.tpl.c84 LIBXSMM_VLA_DECL(2, const element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->…
H A Dlibxsmm_dnn_fusedgroupnorm_st_fwd_custom_f32_bf16_c64_avx512.tpl.c87 LIBXSMM_VLA_DECL(2, element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->da…
H A Dlibxsmm_dnn_fusedbatchnorm_st_fwd_custom_f32_bf16_c64_avx512.tpl.c87 LIBXSMM_VLA_DECL(2, element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->da…
H A Dlibxsmm_dnn_fusedbatchnorm_st_bwd_custom_f32_bf16_c32_avx512.tpl.c91 LIBXSMM_VLA_DECL(2, const element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->…
H A Dlibxsmm_dnn_fusedbatchnorm_st_bwd_custom_f32_bf16_c64_avx512.tpl.c91 LIBXSMM_VLA_DECL(2, const element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->…
H A Dlibxsmm_dnn_fusedgroupnorm_st_bwd_custom_f32_bf16_c64_avx512.tpl.c91 LIBXSMM_VLA_DECL(2, const element_stats_type, brstd, (element_stats_type*)handle->rcpstddev->…