Searched refs:reduce_dataset (Results 1 – 3 of 3) sorted by relevance
72 ReduceDatasetOp reduce_dataset) { in CreateIterator() argument82 reduce_dataset.getLoc(), in CreateIterator()86 builder.create<MakeIteratorOp>(reduce_dataset.getLoc(), in CreateIterator()87 reduce_dataset.input_dataset(), in CreateIterator()95 ReduceDatasetOp reduce_dataset) { in CreateDatasetWhile() argument97 reduce_dataset.getLoc(), in CreateDatasetWhile()105 for (int i = 1; i < reduce_dataset.getNumOperands(); ++i) { in CreateDatasetWhile()106 while_input_values.push_back(reduce_dataset.getOperand(i)); in CreateDatasetWhile()107 while_input_types.push_back(reduce_dataset.getOperand(i).getType()); in CreateDatasetWhile()111 reduce_dataset.getLoc(), while_input_types, /*input=*/while_input_values, in CreateDatasetWhile()[all …]
2681 gen_dataset_ops.reduce_dataset(
7791 * Add `tf.contrib.data.reduce_dataset` which can be used to reduce a