Searched refs:split_image_shape (Results 1 – 1 of 1) sorted by relevance
1396 auto split_image_shape = rewriter.create<UnpackOp>( in matchAndRewrite() local1402 auto batch = split_image_shape.getResult(0); in matchAndRewrite()1403 auto in_y = split_image_shape.getResult(1); in matchAndRewrite()1404 auto in_x = split_image_shape.getResult(2); in matchAndRewrite()1405 auto channels = split_image_shape.getResult(3); in matchAndRewrite()