Searched refs:a4_reshape_shape (Results 1 – 1 of 1) sorted by relevance
971 SmallVector<int64_t, 2> a4_reshape_shape(input_rank); in convertSpaceToBatchNDOp() local974 a4_reshape_shape[0] = batch_size * block_num_elems; in convertSpaceToBatchNDOp()983 a4_reshape_shape[i + 1] = padded_shape[i + 1] / block_shape_val; in convertSpaceToBatchNDOp()988 a4_reshape_shape[1 + block_rank + i] = input_shape[1 + block_rank + i]; in convertSpaceToBatchNDOp()994 rewriter.getI64ArrayAttr(a4_reshape_shape)) in convertSpaceToBatchNDOp()