Searched refs:takewhile_dataset_op (Results 1 – 1 of 1) sorted by relevance
2100 if (auto takewhile_dataset_op = dyn_cast<TakeWhileDatasetOp>(op)) { in InferShapeForSingleOperation() local2102 return InferShapeForTakeWhileDataset(takewhile_dataset_op, max_iterations); in InferShapeForSingleOperation()