Searched refs:tensors_to_save (Results 1 – 2 of 2) sorted by relevance
748 std::vector<std::optional<at::Tensor>>& tensors_to_save, in _get_tensors_to_save() argument780 tensors_to_save.emplace_back(std::nullopt); in _get_tensors_to_save()788 tensors_to_save.emplace_back(tensor); in _get_tensors_to_save()806 const std::vector<std::optional<at::Tensor>>& tensors_to_save, in _save_variables() argument811 size_t num_saved = tensors_to_save.size(); in _save_variables()814 for (const auto& opt_tensor : tensors_to_save) { in _save_variables()1108 std::vector<std::optional<at::Tensor>> tensors_to_save{}; in process_outputs() local1112 tensors_to_save, in process_outputs()1132 _save_variables(tensors_to_save, cdata, grad_fn); in process_outputs()
212 " tensors_to_save = [weight.read_value() for weight in self.model.weights]\n",215 " data=tensors_to_save, name='save')\n",