Searched refs:per_parameter_tensors (Results 1 – 1 of 1) sorted by relevance
96 std::vector<at::Tensor> per_parameter_tensors; in getGradients() local98 per_parameter_tensors.reserve(num_parameters); in getGradients()100 per_parameter_tensors.push_back( in getGradients()104 return per_parameter_tensors; in getGradients()