Searched defs:beta_val (Results 1 – 14 of 14) sorted by relevance
21 auto beta_val = beta.to<scalar_t>(); in addr_kernel_cuda() local48 auto beta_val = beta.to<scalar_t>(); in addr_kernel_cuda() local
17 scalar_t beta_val(beta); in smooth_l1_kernel_cuda() local
465 opmath_t beta_val = beta.to<opmath_t>(); in addmm_out_cuda_impl() local558 opmath_t beta_val = beta.to<opmath_t>(); in baddbmm_out_cuda_impl() local
111 scalar_t beta_val(beta); in smooth_l1_backward_cuda_kernel() local
17 auto beta_val = beta.to<scalar_t>(); in addr_kernel() local46 auto beta_val = beta.to<scalar_t>(); in addr_kernel() local
97 float beta_val(beta); in smooth_l1_backward_cpu_kernel() local144 scalar_t beta_val(beta); in smooth_l1_backward_cpu_kernel() local
765 const scalar_t beta_val(beta); in smooth_l1_kernel() local
750 const opmath_t beta_val = beta_data[c]; in batch_norm_cpu_contiguous_impl() local
56 const auto beta_val = beta.data()[b]; in fuse_batch_normalization_dwc_layer() local90 const auto beta_val = beta.data()[b]; in fuse_batch_normalization_conv_layer() local
61 CTYPE beta_val = utils::scalar_to<CTYPE>(beta); in addmm_out() local
1198 opmath_t beta_val = 0; // bias is added in epilogue in gemm_and_bias() local1418 const float beta_val = 0.0; in scaled_gemm() local1599 int32_t beta_val = 0; in int8_gemm() local
143 const auto beta_val = beta.toComplexDouble(); in addmm_out_sparse_compressed_cuda() local
239 float beta_val = 1.0f; in add_addmm_node() local
261 auto beta_val = beta.toComplexDouble(); in _compressed_row_strided_addmm_out() local