Searched refs:commit_tensor (Results 1 – 4 of 4) sorted by relevance
408 def commit_tensor(self, read_item: ReadItem, tensor: torch.Tensor) -> None: member in LoadPlanner
118 planner.commit_tensor(req, target_tensor)
259 def commit_tensor(self, read_item: ReadItem, tensor: torch.Tensor) -> None: member in DefaultLoadPlanner
670 planner.commit_tensor(req, target_tensor)