Searched refs:_move_dim_to_end (Results 1 – 1 of 1) sorted by relevance
3945 def _move_dim_to_end(tensor, dim_index, rank): function3953 precise_logits = _move_dim_to_end(precise_logits, axis, input_rank)3954 labels = _move_dim_to_end(labels, axis, input_rank)