Home
last modified time | relevance | path

Searched refs:beta_ptr (Results 1 – 21 of 21) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/contrib/cublas/
H A Dcublas.cc245 void *alpha_ptr = nullptr, *beta_ptr = nullptr; in CallGemmEx() local
252 beta_ptr = &beta_int; in CallGemmEx()
255 beta_ptr = &beta_float; in CallGemmEx()
266 beta_ptr, C_data, cuda_out_type, ColumnStride(C), cuda_out_type, algo)); in CallGemmEx()
309 void *alpha_ptr = nullptr, *beta_ptr = nullptr; in CallBatchGemmEx() local
316 beta_ptr = &beta_int; in CallBatchGemmEx()
319 beta_ptr = &beta_float; in CallBatchGemmEx()
329 beta_ptr, C_data, cuda_out_type, ColumnStride3D(C), C_size, batch_size, cuda_out_type, algo)); in CallBatchGemmEx()
/dports/math/libxsmm/libxsmm-1.16.3/src/template/
H A Dlibxsmm_dnn_fusedgroupnorm_st_fwd_custom_generic.tpl.c178 … const element_stats_type* beta_ptr = &LIBXSMM_VLA_ACCESS(2, beta, fm, 0, nFmBlock); variable
194 o = gamma_ptr[v]*(input_f32.f - bmean_ptr[g])*brstd_ptr[g] + beta_ptr[v];
197 o = gamma_ptr[v]*(input_ptr[v] - bmean_ptr[g])*brstd_ptr[g] + beta_ptr[v];
H A Dlibxsmm_dnn_fusedbatchnorm_st_fwd_custom_generic.tpl.c215 … const element_stats_type* beta_ptr = &LIBXSMM_VLA_ACCESS(2, beta, fm, 0, nFmBlock); variable
230 o = gamma_ptr[v]*(input_f32.f - bmean_ptr[v])*brstd_ptr[v] + beta_ptr[v];
233 o = gamma_ptr[v]*(input_ptr[v] - bmean_ptr[v])*brstd_ptr[v] + beta_ptr[v];
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/quantization/mkldnn/
H A Dmkldnn_quantized_batch_norm.cc91 float *beta_ptr = in_data[quantized_batchnorm::kBeta].data().dptr<float>(); in MKLDNNQuantizedBatchNormForward() local
109 (beta_ptr[channel] - moving_mean_ptr[channel] * gamma_ptr[channel] * invstd) * kInt8Range / in MKLDNNQuantizedBatchNormForward()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/quantization/mkldnn/
H A Dmkldnn_quantized_batch_norm.cc91 float *beta_ptr = in_data[quantized_batchnorm::kBeta].data().dptr<float>(); in MKLDNNQuantizedBatchNormForward() local
109 (beta_ptr[channel] - moving_mean_ptr[channel] * gamma_ptr[channel] * invstd) * kInt8Range / in MKLDNNQuantizedBatchNormForward()
/dports/biology/plink/plink-ng-79b2df8c/1.9/
H A Dplink_glm.h43 …integer dgels_lwork, uint32_t standard_beta, double vif_thresh, double* beta_ptr, double* se_ptr, …
46 …param_2d_buf2, float* regression_results, float* sample_1d_buf, double* beta_ptr, double* se_ptr, …
H A Dplink_glm.c8699 …integer dgels_lwork, uint32_t standard_beta, double vif_thresh, double* beta_ptr, double* se_ptr, … in glm_linear_dosage() argument
8803 *beta_ptr = dxx; in glm_linear_dosage()
8810 …param_2d_buf2, float* regression_results, float* sample_1d_buf, double* beta_ptr, double* se_ptr, … in glm_logistic_dosage() argument
8859 *beta_ptr = dxx; in glm_logistic_dosage()
H A Dplink_family.c5135 …ble qt_ssq, double geno_sum, double geno_ssq, double qt_g_prod, double* beta_ptr, double* tstat_pt… in qfam_regress() argument
5201 *beta_ptr = beta; in qfam_regress()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/
H A Dtransformer.cu61 const void *beta_ptr; in CublasStridedBatchedGemm() local
66 beta_ptr = &trueFP16_beta; in CublasStridedBatchedGemm()
70 beta_ptr = &pseudoFP16_beta; in CublasStridedBatchedGemm()
86 beta_ptr, in CublasStridedBatchedGemm()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/
H A Dtransformer.cu61 const void *beta_ptr; in CublasStridedBatchedGemm() local
66 beta_ptr = &trueFP16_beta; in CublasStridedBatchedGemm()
70 beta_ptr = &pseudoFP16_beta; in CublasStridedBatchedGemm()
86 beta_ptr, in CublasStridedBatchedGemm()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/mkldnn/
H A Dmkldnn_conv.cc52 const float *beta_ptr = beta.data().dptr<float>(); in UpdateConvWeightBias() local
68 static_cast<DType>(beta_ptr[c] + alpha * (static_cast<float>(bias_ptr[c]) - mean_ptr[c])); in UpdateConvWeightBias()
70 update_bias_ptr[c] = static_cast<DType>(beta_ptr[c] - alpha * mean_ptr[c]); in UpdateConvWeightBias()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/mkldnn/
H A Dmkldnn_conv.cc52 const float *beta_ptr = beta.data().dptr<float>(); in UpdateConvWeightBias() local
68 static_cast<DType>(beta_ptr[c] + alpha * (static_cast<float>(bias_ptr[c]) - mean_ptr[c])); in UpdateConvWeightBias()
70 update_bias_ptr[c] = static_cast<DType>(beta_ptr[c] - alpha * mean_ptr[c]); in UpdateConvWeightBias()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_uni_batch_normalization.cpp462 Address beta_ptr(size_t offt = 0) { in beta_ptr() function
640 uni_vmovups_maybe_tail(vbeta, beta_ptr(coff)); in forward_channels_nspc_compute()
646 uni_vmovups_maybe_tail(vbeta, beta_ptr(coff)); in forward_channels_nspc_compute()
966 uni_vmovups_maybe_tail(vbeta, beta_ptr()); in forward_channels()
972 uni_vmovups_maybe_tail(vbeta, beta_ptr()); in forward_channels()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_uni_batch_normalization.cpp461 Address beta_ptr(size_t offt = 0) { in beta_ptr() function
639 uni_vmovups_maybe_tail(vbeta, beta_ptr(coff)); in forward_channels_nspc_compute()
645 uni_vmovups_maybe_tail(vbeta, beta_ptr(coff)); in forward_channels_nspc_compute()
965 uni_vmovups_maybe_tail(vbeta, beta_ptr()); in forward_channels()
971 uni_vmovups_maybe_tail(vbeta, beta_ptr()); in forward_channels()
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/common/
H A Ddnn_common.h2131 …const float* beta_ptr, const float* gamma_ptr, float* expectval_ptr, float* rcpstddev_ptr, float* … in naive_fusedbatchnorm_fp() argument
2199 float o = gamma_ptr[fm]*(input_val - expectval_ptr[fm])*rcpstddev_ptr[fm] + beta_ptr[fm]; in naive_fusedbatchnorm_fp()
2216 … const float* beta_ptr, float* del_beta_ptr, const float* gamma_ptr, float* del_gamma_ptr, in naive_fusedbatchnorm_bp() argument
2237 LIBXSMM_UNUSED(beta_ptr); in naive_fusedbatchnorm_bp()
2292 …const float* beta_ptr, const float* gamma_ptr, float* expectval_ptr, float* rcpstddev_ptr, float* … in naive_fusedgroupnorm_fp() argument
2363 …r[g*nFMG+fmg]*(input_val - expectval_ptr[img*nG+g])*rcpstddev_ptr[img*nG+g] + beta_ptr[g*nFMG+fmg]; in naive_fusedgroupnorm_fp()
2381 … const float* beta_ptr, float* del_beta_ptr, const float* gamma_ptr, float* del_gamma_ptr, in naive_fusedgroupnorm_bp() argument
2410 LIBXSMM_UNUSED(beta_ptr); in naive_fusedgroupnorm_bp()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/
H A Dlinalg_impl.h285 const void *beta_ptr; local
290 beta_ptr = &trueFP16_beta;
294 beta_ptr = &pseudoFP16_beta;
305 beta_ptr,
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/
H A Dlinalg_impl.h285 const void *beta_ptr; local
290 beta_ptr = &trueFP16_beta;
294 beta_ptr = &pseudoFP16_beta;
305 beta_ptr,
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/
H A Djit_uni_batch_normalization.cpp559 XReg beta_ptr(size_t offt = 0) { in beta_ptr() function
771 uni_load_maybe_tail(vbeta, beta_ptr(coff)); in forward_channels_nspc_compute()
1122 uni_load_maybe_tail(vbeta, beta_ptr()); in forward_channels()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/
H A Djit_uni_batch_normalization.cpp559 XReg beta_ptr(size_t offt = 0) { in beta_ptr() function
771 uni_load_maybe_tail(vbeta, beta_ptr(coff)); in forward_channels_nspc_compute()
1122 uni_load_maybe_tail(vbeta, beta_ptr()); in forward_channels()
/dports/biology/gemma/GEMMA-0.98.3/src/
H A Dvc.cpp428 string beta_ptr[] = {"beta", "BETA", "b", "B"}; in ReadHeader_vc() local
429 set<string> beta_set(beta_ptr, beta_ptr + 4); in ReadHeader_vc()
H A Dgemma_io.cpp2370 string beta_ptr[] = {"beta", "BETA", "b", "B"}; in ReadHeader_io() local
2371 set<string> beta_set(beta_ptr, beta_ptr + 4); in ReadHeader_io()