Searched refs:remaining_shape_rank (Results 1 – 2 of 2) sorted by relevance
829 int remaining_shape_rank = input_rank - block_rank - 1; in convertSpaceToBatchNDOp() local876 for (int i = 0; i < remaining_shape_rank; i++) { in convertSpaceToBatchNDOp()904 SmallVector<int64_t, 2> a2_shape(1 + block_rank * 2 + remaining_shape_rank); in convertSpaceToBatchNDOp()920 for (int i = 0; i < remaining_shape_rank; i++) { in convertSpaceToBatchNDOp()987 for (int i = 0; i < remaining_shape_rank; i++) { in convertSpaceToBatchNDOp()1080 int remaining_shape_rank = input_rank - block_rank - 1; in convertBatchToSpaceNDOp() local1165 for (int i = 0; i < remaining_shape_rank; i++) { in convertBatchToSpaceNDOp()1194 for (int i = 0; i < remaining_shape_rank; i++) { in convertBatchToSpaceNDOp()
242 size_t remaining_shape_rank = %input.rank - %block.rank - 1715 size_t remaining_shape_rank = %input.rank - block_rank - 1;733 vector <size_t> a2_shape(1 + block_rank * 2 + remaining_shape_rank)740 for (int i = 0; i < remaining_shape_rank; i++) {