Searched defs:FuseParallelMapAndBatch (Results 1 – 1 of 1) sorted by relevance
27 struct FuseParallelMapAndBatch : public OpRewritePattern<BatchDatasetV2Op> { struct30 LogicalResult matchAndRewrite(BatchDatasetV2Op op, in matchAndRewrite()