| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | IGammaKernel.cu | 32 using accscalar_t = at::acc_type<scalar_t, /*is_cuda=*/true>; in ratevl() typedef
83 using accscalar_t = at::acc_type<scalar_t, /*is_cuda=*/true>; in lanczos_sum_expg_scaled() typedef
127 using accscalar_t = at::acc_type<scalar_t, /*is_cuda=*/true>; in _igam_helper_fac() typedef
160 using accscalar_t = at::acc_type<scalar_t, /*is_cuda=*/true>; in _igam_helper_series() typedef
194 using accscalar_t = at::acc_type<scalar_t, /*is_cuda=*/true>; in _igamc_helper_series() typedef
221 using accscalar_t = at::acc_type<scalar_t, /*is_cuda=*/true>; in _igam_helper_asymptotic_series() typedef
313 using accscalar_t = at::acc_type<scalar_t, /*is_cuda=*/true>; in _igamc_helper_continued_fraction() typedef
385 using accscalar_t = at::acc_type<scalar_t, /*is_cuda=*/true>; in calc_igammac() typedef
468 using accscalar_t = at::acc_type<scalar_t, /*is_cuda=*/true>; in calc_igamma() typedef
|
| H A D | MultiLabelMarginCriterion.cu | 231 using accscalar_t = at::acc_type<scalar_t, true>; in multilabel_margin_loss_forward_out_cuda_template() typedef
256 using accscalar_t = at::acc_type<scalar_t, true>; in multilabel_margin_loss_forward_out_cuda_template() typedef
282 using accscalar_t = at::acc_type<scalar_t, true>; in multilabel_margin_loss_forward_out_cuda_template() typedef
341 using accscalar_t = at::acc_type<scalar_t, true>; in multilabel_margin_loss_backward_cuda_out_template() typedef
370 using accscalar_t = at::acc_type<scalar_t, true>; in multilabel_margin_loss_backward_cuda_out_template() typedef
|
| H A D | Distributions.cu | 80 using accscalar_t = at::acc_type<scalar_t, true>; in binomial_cuda_kernel() typedef
101 using accscalar_t = at::acc_type<scalar_t, true>; in gamma_cuda_kernel() typedef
190 using accscalar_t = at::acc_type<scalar_t, true>; in launch_standard_gamma_grad_kernel() typedef
200 using accscalar_t = at::acc_type<scalar_t, true>; in launch_dirichlet_grad_kernel() typedef
|
| H A D | Normalization.cu | 107 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_elementwise() typedef
187 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_elementwise_backward_train() typedef
238 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_elementwise_backward_train() typedef
279 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_elementwise_backward_eval() typedef
296 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_elementwise_backward_eval() typedef
587 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_backward_cuda() typedef
716 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_gather_stats_with_counts_cuda() typedef
744 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_backward_reduce_cuda() typedef
779 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_backward_elemt_cuda() typedef
|
| H A D | Normalization.cuh | 428 const GenericPackedTensorAccessor<accscalar_t, 2, RestrictPtrTraits, index_t> vec_mean, in batch_norm_reduce_statistics_kernel()
607 using accscalar_t = at::acc_type<stat_scalar_t, true>; in batch_norm_backward_cuda_template() typedef
664 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_stats_cuda_template() typedef
1428 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_stats_channels_last_cuda_template() typedef
1488 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_elemt_channels_last_cuda_template() typedef
1509 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_elemt_channels_last_cuda_template() typedef
1565 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_backward_reduce_cuda_channels_last_template() typedef
1590 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_backward_reduce_cuda_channels_last_template() typedef
1637 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_backward_elemt_channels_last_cuda_template() typedef
1660 using accscalar_t = at::acc_type<scalar_t, true>; in batch_norm_backward_elemt_channels_last_cuda_template() typedef
[all …]
|
| H A D | WeightNorm.cu | 372 using accscalar_t = acc_type<scalar_t, true>; in weight_norm_cuda() typedef
403 using accscalar_t = acc_type<scalar_t, true>; in weight_norm_cuda() typedef
461 using accscalar_t = acc_type<scalar_t, true>; in weight_norm_backward_cuda() typedef
494 using accscalar_t = acc_type<scalar_t, true>; in weight_norm_backward_cuda() typedef
|
| H A D | SoftMax.cu | 845 using accscalar_t = acc_type<scalar_t, true>; in host_softmax() typedef
927 using accscalar_t = acc_type<scalar_t, true>; in host_softmax() typedef
981 using accscalar_t = acc_type<scalar_t, true>; in host_softmax_backward() typedef
1036 using accscalar_t = acc_type<scalar_t, true>; in host_softmax_backward() typedef
1176 using accscalar_t = acc_type<scalar_t, true>; in masked_softmax_cuda() typedef
1196 using accscalar_t = acc_type<scalar_t, true>; in masked_softmax_cuda() typedef
1258 using accscalar_t = acc_type<scalar_t, true>; in masked_softmax_backward_cuda() typedef
|
| H A D | UpSampleBilinear2d.cu | 292 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_bilinear2d_out_cuda_template() typedef
338 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_bilinear2d_out_cuda_template() typedef
405 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_bilinear2d_backward_out_cuda_template() typedef
433 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_bilinear2d_backward_out_cuda_template() typedef
707 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_gen2d_aa_out_cuda_template() typedef
811 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_gen2d_aa_backward_out_cuda_template() typedef
|
| H A D | Embedding.cu | 139 using accscalar_t = acc_type<scalar_t, true>; in embedding_backward_kernel() typedef
276 using accscalar_t = acc_type<scalar_t, true>; in embedding_dense_backward_cuda() typedef
380 using accscalar_t = acc_type<scalar_t, true>; in embedding_renorm_cuda_() typedef
|
| H A D | PointwiseOpsKernel.cu | 51 using accscalar_t = at::acc_type<scalar_t, true>; in addcmul_cuda_kernel() typedef
99 using accscalar_t = at::acc_type<scalar_t, true>; in addcdiv_cuda_kernel() typedef
|
| H A D | UpSampleNearest2d.cu | 309 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_nearest2d_out_cuda_template() typedef
381 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_nearest2d_backward_out_cuda_template() typedef
417 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_nearest2d_backward_out_cuda_template() typedef
|
| H A D | RangeFactories.cu | 179 using accscalar_t = at::acc_type<scalar_t, true>; in range_cuda_out() typedef
215 using accscalar_t = at::acc_type<scalar_t, true>; in arange_cuda_out() typedef
|
| H A D | RNN.cu | 374 using accscalar_t = acc_type<scalar_t, /*is_cuda=*/true>; in lstm_forward_impl() typedef
411 using accscalar_t = acc_type<scalar_t, /*is_cuda=*/true>; in lstm_backward_impl() typedef
447 using accscalar_t = acc_type<scalar_t, /*is_cuda=*/true>; in gru_forward_impl() typedef
481 using accscalar_t = acc_type<scalar_t, /*is_cuda=*/true>; in gru_backward_impl() typedef
|
| H A D | EmbeddingBag.cu | 122 using accscalar_t = acc_type<scalar_t, true>; in EmbeddingBag_updateOutputKernel_sum_mean() typedef
245 using accscalar_t = acc_type<scalar_t, true>; in EmbeddingBag_accGradParametersKernel_max() typedef
480 using accscalar_t = acc_type<scalar_t, true>; in _embedding_bag_per_sample_weights_backward_kernel() typedef
|
| H A D | UpSampleNearest1d.cu | 133 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_nearest1d_out_cuda_template() typedef
184 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_nearest1d_backward_out_cuda_template() typedef
|
| H A D | UpSampleLinear1d.cu | 144 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_linear1d_out_cuda_template() typedef
188 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_linear1d_backward_out_cuda_template() typedef
|
| H A D | AveragePool3d.cu | 404 using accscalar_t = acc_type<scalar_t, true>; in TORCH_IMPL_FUNC() typedef
527 using accscalar_t = acc_type<scalar_t, true>; in TORCH_IMPL_FUNC() typedef
563 using accscalar_t = acc_type<scalar_t, true>; in TORCH_IMPL_FUNC() typedef
|
| H A D | UpSampleBicubic2d.cu | 196 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_bicubic2d_out_cuda_template() typedef
252 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_bicubic2d_backward_out_cuda_template() typedef
|
| H A D | Dropout.cu | 236 using accscalar_t = acc_type<scalar_t, true>; in launcher() typedef
399 using accscalar_t = acc_type<scalar_t, true>; in dropout_backward_cuda() typedef
|
| H A D | UpSampleNearest3d.cu | 189 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_nearest3d_out_cuda_template() typedef
265 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_nearest3d_backward_out_cuda_template() typedef
|
| H A D | UpSampleTrilinear3d.cu | 270 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_trilinear3d_out_cuda_template() typedef
340 using accscalar_t = at::acc_type<scalar_t, true>; in upsample_trilinear3d_backward_out_cuda_template() typedef
|
| H A D | BinaryDivTruncKernel.cu | 32 using accscalar_t = at::acc_type<scalar_t, true>; in div_trunc_kernel_cuda() typedef
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Distributions.h | 90 C10_DEVICE scalar_t sample_gamma(scalar_t alpha, BaseSampler<accscalar_t, uniform_sampler_t>& stand… in sample_gamma()
151 C10_DEVICE scalar_t binomial_inversion(scalar_t count, scalar_t prob, BaseSampler<accscalar_t, unif… in binomial_inversion()
171 C10_DEVICE scalar_t btrs(scalar_t count, scalar_t prob, BaseSampler<accscalar_t, uniform_sampler_t>… in btrs()
224 C10_DEVICE scalar_t sample_binomial(scalar_t count, scalar_t prob, BaseSampler<accscalar_t, uniform… in sample_binomial()
|
| H A D | RangeFactories.cpp | 155 using accscalar_t = at::acc_type<scalar_t, false>; in range_out() typedef
198 using accscalar_t = at::acc_type<scalar_t, false>; in arange_out() typedef
|
| H A D | LossMultiLabelMargin.cpp | 31 using accscalar_t = at::acc_type<scalar_t, false>; in multilabel_margin_loss_forward_inner_sum_cpu() typedef
69 using accscalar_t = at::acc_type<scalar_t, false>; in multilabel_margin_loss_forward_out_frame() typedef
|