Searched refs:split_image_shape (Results 1 – 1 of 1) sorted by relevance
1368 auto split_image_shape = rewriter.create<UnpackOp>( in matchAndRewrite() local1374 auto batch = split_image_shape.getResult(0); in matchAndRewrite()1375 auto in_y = split_image_shape.getResult(1); in matchAndRewrite()1376 auto in_x = split_image_shape.getResult(2); in matchAndRewrite()1377 auto channels = split_image_shape.getResult(3); in matchAndRewrite()