Searched defs:tensor_vec (Results 1 – 3 of 3) sorted by relevance
66 void save(const std::vector<torch::Tensor>& tensor_vec, SaveToArgs&&... args) { in save()128 void load(std::vector<torch::Tensor>& tensor_vec, LoadFromArgs&&... args) { in load()
465 auto tensor_vec = tensors.vec(); in allreduce_sparse_cuda_() local
210 auto tensor_vec = ival.toTensorVector(); in getTensor() local