Searched defs:_split_tensor (Results 1 – 1 of 1) sorted by relevance
69 def _split_tensor( member in Shard444 def _split_tensor( member in _StridedShard