Searched refs:inner_rank (Results 1 – 3 of 3) sorted by relevance
475 def squeeze_batch_dims(inp, op, inner_rank): argument495 inner_shape = shape[-inner_rank:]497 inner_shape = array_ops.shape(inp)[-inner_rank:]499 batch_shape = shape[:-inner_rank]501 batch_shape = array_ops.shape(inp)[:-inner_rank]511 out_inner_shape = out_reshaped.shape[-inner_rank:]513 out_inner_shape = array_ops.shape(out_reshaped)[-inner_rank:]518 out.set_shape(inp.shape[:-inner_rank] + out.shape[-inner_rank:])
277 def squeeze_batch_dims(inp, op, inner_rank, name=None): argument299 inner_shape = shape[-inner_rank:]301 inner_shape = array_ops.shape(inp)[-inner_rank:]303 batch_shape = shape[:-inner_rank]305 batch_shape = array_ops.shape(inp)[:-inner_rank]315 out_inner_shape = out_reshaped.shape[-inner_rank:]317 out_inner_shape = array_ops.shape(out_reshaped)[-inner_rank:]322 out.set_shape(inp.shape[:-inner_rank] + out.shape[-inner_rank:])1905 inner_rank=3,2290 inner_rank=3,[all …]
267 outputs, _apply_fn, inner_rank=self.rank + 1)