Home
last modified time | relevance | path

Searched refs:finalize_func (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dgroup_by_reducer_test.py44 finalize_func=lambda x: x)
63 finalize_func=lambda x, _: x)
79 finalize_func=lambda x: x)
101 finalize_func=lambda x: x.values[0])
123 finalize_func=lambda x, y: (x, y))
143 finalize_func=lambda x: x)
158 finalize_func=lambda x: x)
172 finalize_func=lambda x: x)
Doptimize_dataset_test.py69 finalize_func=lambda _: var)
/external/tensorflow/tensorflow/core/kernels/data/
Dgenerator_dataset_op.cc51 std::unique_ptr<CapturedFunction> finalize_func, in Dataset() argument
57 finalize_func_(std::move(finalize_func)), in Dataset()
217 std::unique_ptr<CapturedFunction> finalize_func; in MakeDataset() local
220 kFinalizeFuncOtherArgs, &finalize_func)); in MakeDataset()
224 std::move(finalize_func), output_types_, output_shapes_); in MakeDataset()
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dgrouping.py281 self._make_finalize_func(reducer.finalize_func)
291 finalize_func=self._finalize_func.function,
376 def _make_finalize_func(self, finalize_func): argument
379 finalize_func, self._transformation_name(),
477 def __init__(self, init_func, reduce_func, finalize_func): argument
480 self._finalize_func = finalize_func
491 def finalize_func(self): member in Reducer
Dprefetching_ops.py210 finalize_func=self._finalize_func,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.-reducer.pbtxt6 name: "finalize_func"
19 …argspec: "args=[\'self\', \'init_func\', \'reduce_func\', \'finalize_func\'], varargs=None, keywor…
Dtensorflow.raw_ops.pbtxt1437 …nc_other_arguments\', \'key_func\', \'init_func\', \'reduce_func\', \'finalize_func\', \'output_ty…
1777 …args\', \'finalize_func_other_args\', \'init_func\', \'next_func\', \'finalize_func\', \'output_ty…
1801 …nc_other_arguments\', \'key_func\', \'init_func\', \'reduce_func\', \'finalize_func\', \'output_ty…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.-reducer.pbtxt6 name: "finalize_func"
19 …argspec: "args=[\'self\', \'init_func\', \'reduce_func\', \'finalize_func\'], varargs=None, keywor…
Dtensorflow.raw_ops.pbtxt1437 …nc_other_arguments\', \'key_func\', \'init_func\', \'reduce_func\', \'finalize_func\', \'output_ty…
1777 …args\', \'finalize_func_other_args\', \'init_func\', \'next_func\', \'finalize_func\', \'output_ty…
1801 …nc_other_arguments\', \'key_func\', \'init_func\', \'reduce_func\', \'finalize_func\', \'output_ty…
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ExperimentalGroupByReducerDataset.pbtxt56 building a closure for `finalize_func`.
60 name: "finalize_func"
Dapi_def_GroupByReducerDataset.pbtxt56 building a closure for `finalize_func`.
60 name: "finalize_func"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DGeneratorDataset.pbtxt28 name: "finalize_func"
DGroupByReducerDataset.pbtxt40 name: "finalize_func"
DExperimentalGroupByReducerDataset.pbtxt40 name: "finalize_func"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DGeneratorDataset.pbtxt28 name: "finalize_func"
DGroupByReducerDataset.pbtxt40 name: "finalize_func"
DExperimentalGroupByReducerDataset.pbtxt40 name: "finalize_func"
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test.py471 finalize_func = lambda x: True function
474 (stateful_key_func, init_func, reduce_func, finalize_func),
475 (key_func, stateful_init_func, reduce_func, finalize_func),
476 (key_func, init_func, stateful_reduce_func, finalize_func),
479 for key_func, init_func, reduce_func, finalize_func in test_cases:
481 reducer = grouping.Reducer(init_func, reduce_func, finalize_func)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dgroup_by_reducer_serialization_test.py39 finalize_func=lambda x: x)
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dgroup_by_reducer_dataset_op.cc161 AttrValue finalize_func; in AsGraphDefInternal() local
162 b->BuildAttrValue(captured_finalize_func_->func(), &finalize_func); in AsGraphDefInternal()
186 {"finalize_func", finalize_func}, in AsGraphDefInternal()
/external/tensorflow/tensorflow/python/data/ops/
Dmulti_device_iterator_ops.py138 finalize_func=self._finalize_func,
180 finalize_func=self._finalize_func,
Ddataset_ops.py3631 def __init__(self, init_args, init_func, next_func, finalize_func, argument
3664 finalize_func,
3677 finalize_func=self._finalize_func.function,
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt14901 name: "finalize_func"
18399 name: "finalize_func"
18588 name: "finalize_func"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt23365 name: "finalize_func"
27919 name: "finalize_func"