Searched refs:output_axes (Results 1 – 1 of 1) sorted by relevance
429 def output_axes(axes): function443 axes = output_axes(labeled_tensors[i].axes)