Searched defs:write_tensor (Results 1 – 2 of 2) sorted by relevance
181 def write_tensor(self, name: str, t: torch.Tensor) -> None: member in ContentStoreWriter
190 def write_tensor(fp, name, x): function