Searched refs:InferShapeForTakeWhileDataset (Results 1 – 1 of 1) sorted by relevance
767 bool InferShapeForTakeWhileDataset(TakeWhileDatasetOp op,1133 bool ShapeInference::InferShapeForTakeWhileDataset(TakeWhileDatasetOp op, in InferShapeForTakeWhileDataset() function in mlir::TF::ShapeInference2102 return InferShapeForTakeWhileDataset(takewhile_dataset_op, max_iterations); in InferShapeForSingleOperation()