Searched refs:get_param_from_index (Results 1 – 2 of 2) sorted by relevance
556 at::Tensor& get_param_from_index(size_t index);
660 auto& variable = get_param_from_index(index); in autograd_hook()791 at::Tensor& Reducer::get_param_from_index(size_t index) { in get_param_from_index() function in c10d::Reducer