Searched defs:inner_rank (Results 1 – 2 of 2) sorted by relevance
475 def squeeze_batch_dims(inp, op, inner_rank): argument
277 def squeeze_batch_dims(inp, op, inner_rank, name=None): argument