Searched defs:inner_rank (Results 1 – 3 of 3) sorted by relevance
471 def squeeze_batch_dims(inp, op, inner_rank): argument
739 def inner_rank(self): member in DynamicRaggedShape791 def _with_inner_rank(self, inner_rank): argument1351 def inner_rank(self) -> Optional[int]: member in DynamicRaggedShape.Spec
428 def squeeze_batch_dims(inp, op, inner_rank, name=None): argument