Searched defs:dataset_while (Results 1 – 1 of 1) sorted by relevance
110 auto dataset_while = builder.create<TF::WhileRegionOp>( in CreateDatasetWhile() local126 void PopulateDatasetWhileCond(OpBuilder builder, WhileRegionOp dataset_while, in PopulateDatasetWhileCond()216 FuncOp reduce_func, WhileRegionOp dataset_while, in PopulateDatasetWhileBody()305 WhileRegionOp dataset_while = CreateDatasetWhile(builder, reduce_dataset); in DecomposeReduceDatasetInFunction() local