Searched refs:resolve_tensor (Results 1 – 4 of 4) sorted by relevance
398 def resolve_tensor(self, read_item: ReadItem) -> torch.Tensor: member in LoadPlanner
112 target_tensor = planner.resolve_tensor(req).detach()
255 def resolve_tensor(self, read_item: ReadItem): member in DefaultLoadPlanner
664 target_tensor = planner.resolve_tensor(req).detach()