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