Searched defs: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() local
290 std::vector<at::Tensor> tensors_to_serialize = { in __getstate__() local429 std::vector<at::Tensor> tensors_to_serialize{ in __getstate__() local