Searched refs:tensors_to_serialize (Results 1 – 2 of 2) sorted by relevance
525 std::vector<at::Tensor> tensors_to_serialize = {std::move(weight)}; in register_embedding_params() local532 std::move(tensors_to_serialize), in register_embedding_params()
290 std::vector<at::Tensor> tensors_to_serialize = { in __getstate__() local298 tensors_to_serialize, in __getstate__()429 std::vector<at::Tensor> tensors_to_serialize{ in __getstate__() local440 tensors_to_serialize, in __getstate__()