Home
last modified time | relevance | path

Searched defs:staging_data (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DNormalization.cuh948 volatile accscalar_t* staging_data, in batch_norm_collect_statistics_channels_last_kernel()
1176 volatile accscalar_t* staging_data, in batch_norm_backward_reduce_channels_last_kernel()
1444 at::Tensor staging_data; in batch_norm_stats_channels_last_cuda_template() local
1555 at::Tensor staging_data; in batch_norm_backward_reduce_cuda_channels_last_template() local