Searched defs:set_tensors (Results 1 – 2 of 2) sorted by relevance
238 def set_tensors(self, names: Iterable[str], values: Iterable[torch.Tensor]) -> None: member in NamedMemberAccessor
365 virtual void set_tensors(std::vector<at::Tensor>) {} in set_tensors() function