Searched defs:finalize_func (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | grouping.py | 376 def _make_finalize_func(self, finalize_func): argument
477 def __init__(self, init_func, reduce_func, finalize_func): argument
491 def finalize_func(self): member in Reducer
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | generator_dataset_op.cc | 51 std::unique_ptr<CapturedFunction> finalize_func, in Dataset()
217 std::unique_ptr<CapturedFunction> finalize_func; in MakeDataset() local
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | checkpoint_test.py | 471 finalize_func = lambda x: True function
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | group_by_reducer_dataset_op.cc | 161 AttrValue finalize_func; in AsGraphDefInternal() local
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 3631 def __init__(self, init_args, init_func, next_func, finalize_func, argument
|