Searched refs:a2_reshape_a1_rank (Results 1 – 1 of 1) sorted by relevance
937 int32_t a2_reshape_a1_rank = in convertSpaceToBatchNDOp() local939 SmallVector<int32_t, 8> a3_perm(a2_reshape_a1_rank); in convertSpaceToBatchNDOp()940 SmallVector<int64_t, 2> a3_transpose_shape(a2_reshape_a1_rank); in convertSpaceToBatchNDOp()947 for (int i = 1 + block_rank * 2; i < a2_reshape_a1_rank; i++) { in convertSpaceToBatchNDOp()