Searched refs:transpose_permutation (Results 1 – 2 of 2) sorted by relevance
67 llvm::SmallVector<int64_t, 5> transpose_permutation(left_dims.begin(), in TransposeReshape() local69 transpose_permutation.append(right_dims.begin(), right_dims.end()); in TransposeReshape()72 {static_cast<int64_t>(transpose_permutation.size())}, in TransposeReshape()77 llvm::makeArrayRef(transpose_permutation)) in TransposeReshape()82 for (auto val : transpose_permutation) { in TransposeReshape()
432 transpose_permutation = permutation + [435 tile_assignment = _np.transpose(device_mesh, transpose_permutation)