Searched refs:InferShapeForMapDataset (Results 1 – 1 of 1) sorted by relevance
759 bool InferShapeForMapDataset(MapDatasetOp op, int64_t max_iterations);1119 bool ShapeInference::InferShapeForMapDataset(MapDatasetOp op, in InferShapeForMapDataset() function in mlir::TF::ShapeInference2097 return InferShapeForMapDataset(map_dataset_op, max_iterations); in InferShapeForSingleOperation()