Searched refs:InferShapeForReduceDataset (Results 1 – 1 of 1) sorted by relevance
763 bool InferShapeForReduceDataset(ReduceDatasetOp op, int64_t max_iterations);1147 bool ShapeInference::InferShapeForReduceDataset(ReduceDatasetOp op, in InferShapeForReduceDataset() function in mlir::TF::ShapeInference2046 return InferShapeForReduceDataset(reduce_dataset_op, max_iterations); in InferShapeForSingleOperation()