Searched defs:bucket_view (Results 1 – 1 of 1) sorted by relevance
309 const at::Tensor& bucket_view) { in check_grad_layout()352 auto& bucket_view = bucket.bucket_views_in[bucket_index.intra_bucket_index]; in mark_variable_ready_dense() local1268 auto& bucket_view = bucket.bucket_views_in.back(); in initialize_bucket_views() local1463 const auto& bucket_view = bucket.bucket_views_out[intra_bucket_index]; in copy_bucket_to_grad() local