Searched defs:scale_ptr (Results 1 – 8 of 8) sorted by relevance
189 float* scale_ptr, in _calc_moving_avg_qparams_helper()294 float* scale_ptr = scale.data_ptr<float>(); in fused_moving_avg_obs_fake_quant_cuda() local
63 float* scale_ptr = scale.data_ptr<float>(); in fake_quantize_tensor_cachemask_tensor_qparams_kernel_cuda() local
159 const CTYPE_PARAMS* scale_ptr = scales + qparams_index; in embedding_byte_per_channel() local
199 const CTYPE_PARAMS* scale_ptr = scales + qparams_index; in embedding_xbit_per_channel() local
166 const __nv_bfloat16* scale_ptr = reinterpret_cast<const __nv_bfloat16*>(&scale_frag); in dequantize() local
234 const opmath_t* scale_ptr, in ApplyScaleBias()253 const opmath_t* scale_ptr, in ApplyScaleBias()361 opmath_t* scale_ptr = buffer_data + i * 2 * D; in GroupNormKernelImplChannelsLastInternal() local449 opmath_t* scale_ptr = buffer_data + n * 2 * C; in GroupNormKernelImplChannelsLastInternal() local475 opmath_t* scale_ptr = buffer_data + n * 2 * C; in GroupNormKernelImplChannelsLastInternal() local
915 auto scale_ptr = StreamExecutorUtil::AsDeviceMemory<U>(scale); in operator ()() local1112 auto scale_ptr = StreamExecutorUtil::AsDeviceMemory<U>(scale); in operator ()() local
3146 float* scale_ptr = buffer_data + grpIdx * 2 * channels_per_group; local3257 float* scale_ptr = buffer_data + n * 2 * C; local3280 float* scale_ptr = buffer_data + n * 2 * C; local