Searched refs:_unwrap_or_tile (Results 1 – 1 of 1) sorted by relevance
1366 *[self._unwrap_or_tile(w) for w in outputs])1386 def _unwrap_or_tile(self, wrapped_tensor): member in PFor1413 return self._unwrap_or_tile(output)1453 batched_args.append(self._unwrap_or_tile(batched_arg))