Searched defs:reduce_dataset (Results 1 – 1 of 1) sorted by relevance
72 ReduceDatasetOp reduce_dataset) { in CreateIterator()95 ReduceDatasetOp reduce_dataset) { in CreateDatasetWhile()141 OpBuilder builder, ReduceDatasetOp reduce_dataset, FuncOp reduce_func, in CreateOptionalDatasetIf()215 void PopulateDatasetWhileBody(OpBuilder builder, ReduceDatasetOp reduce_dataset, in PopulateDatasetWhileBody()269 auto decompose_result = function.walk([&](ReduceDatasetOp reduce_dataset) { in DecomposeReduceDatasetInFunction()