Searched defs:staging_data (Results 1 – 1 of 1) sorted by relevance
948 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() local1555 at::Tensor staging_data; in batch_norm_backward_reduce_cuda_channels_last_template() local