Searched defs:scalar_val (Results 1 – 12 of 12) sorted by relevance
74 at::opmath_type<f_inputs_type> scalar_val = 0, in jitted_gpu_kernel()167 auto scalar_val = iter.scalar_value<opmath_t>(1); in opmath_jitted_gpu_kernel_with_scalars() local177 auto scalar_val = iter.scalar_value<opmath_t>(2); in opmath_jitted_gpu_kernel_with_scalars() local
88 void* scalar_val, in launch_jitted_unrolled_kernel()116 void *scalar_val, c10::ArrayRef<void*> extra_args) { in launch_jitted_vectorized_kernel()184 void *scalar_val) { in jitted_gpu_kernel_generic()267 at::opmath_type<f_inputs_type> scalar_val, in jitted_gpu_kernel_impl()
211 opmath_arg_t scalar_val{}; in opmath_symmetric_gpu_kernel_with_scalars() local
37 scalar_t scalar_val = value.to<scalar_t>(); in addcmul_cpu_kernel() local77 scalar_t scalar_val = value.to<scalar_t>(); in addcdiv_cpu_kernel() local221 scalar_t scalar_val = value.to<scalar_t>(); in mse_backward_cpu_kernel() local
336 scalar_t scalar_val = value.to<scalar_t>(); in masked_fill_kernel() local
74 float scalar_val = 0; in launch_jitted_vectorized_kernel_dynamic() local162 float scalar_val = 0; in launch_jitted_unrolled_kernel_dynamic() local
76 CTYPE_SCALAR scalar_val = *scalar->const_data_ptr<CTYPE_SCALAR>(); in opt_div_out() local
114 CTYPE_SCALAR scalar_val = *scalar->const_data_ptr<CTYPE_SCALAR>(); in opt_sub_out() local
248 struct vtn_ssa_value *scalar_val = vtn_ssa_value(b, w[4]); in vtn_handle_cooperative_alu() local
1027 const auto scalar_val = extract_vals_from_wrapped_scalar(zero_dim); in _mul_sparse_sparse_zero_dim_out() local1032 const auto scalar_val = extract_vals_from_wrapped_scalar(other); in _mul_sparse_sparse_zero_dim_out() local1038 const auto scalar_val = extract_vals_from_wrapped_scalar(zero_dim.coalesce()); in _mul_sparse_sparse_zero_dim_out() local
125 Value scalar_val = CreateI32SplatConst(loc, rewriter, {}, val); in CreateI32SplatTensor() local
1479 int scalar_val = i - 3.0f; in TEST() local