Searched refs:_swap_axis (Results 1 – 1 of 1) sorted by relevance
3602 def _swap_axis(input_tensor, dim_index, last_index, name=None): function3645 inputs = _swap_axis(inputs, dim_axis, math_ops.subtract(input_rank, 1))3649 output = _swap_axis(