Searched refs:InferShapeForMapDataset (Results 1 – 1 of 1) sorted by relevance
737 bool InferShapeForMapDataset(MapDatasetOp op);937 bool ShapeInference::InferShapeForMapDataset(MapDatasetOp op) { in InferShapeForMapDataset() function in mlir::TF::ShapeInference1298 return InferShapeForMapDataset(map_dataset_op); in InferShapeForSingleOperation()