Searched refs:_swap_axis (Results 1 – 1 of 1) sorted by relevance
2816 def _swap_axis(logits, dim_index, last_index, name=None): function2859 logits = _swap_axis(logits, dim_axis, math_ops.subtract(input_rank, 1))2864 output = _swap_axis(