Searched refs:a3_transpose_shape (Results 1 – 1 of 1) sorted by relevance
940 SmallVector<int64_t, 2> a3_transpose_shape(a2_reshape_a1_rank); in convertSpaceToBatchNDOp() local951 for (int i = 0; i < a3_transpose_shape.size(); i++) { in convertSpaceToBatchNDOp()952 a3_transpose_shape[i] = a2_shape[a3_perm[i]]; in convertSpaceToBatchNDOp()960 RankedTensorType::get(ArrayRef<int64_t>(a3_transpose_shape), in convertSpaceToBatchNDOp()