Searched refs:permuted_axes (Results 1 – 2 of 2) sorted by relevance
715 permuted_axes = [labeled_tensor.axes[n] for n in axis_order]717 return LabeledTensor(transpose_tensor, permuted_axes)
877 const Dims<4>& output_dims, const int* permuted_axes) { in Transpose() argument881 params.perm[i] = 3 - permuted_axes[3 - i]; in Transpose()