Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dgroup_by_reducer_test.py43 finalize_func=lambda x: x)
61 finalize_func=lambda x, _: x)
76 finalize_func=lambda x: x)
97 finalize_func=lambda x: x.values[0])
118 finalize_func=lambda x, y: (x, y))
137 finalize_func=lambda x: x)
151 finalize_func=lambda x: x)
164 finalize_func=lambda x: x)
/external/tensorflow/tensorflow/core/kernels/data/
Dgenerator_dataset_op.cc35 std::unique_ptr<CapturedFunction> finalize_func, in Dataset() argument
41 finalize_func_(std::move(finalize_func)), in Dataset()
174 std::unique_ptr<CapturedFunction> finalize_func; in MakeDataset() local
177 &finalize_func)); in MakeDataset()
181 std::move(finalize_func), output_types_, output_shapes_); in MakeDataset()
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dgrouping.py255 self._make_finalize_func(reducer.finalize_func)
265 finalize_func=self._finalize_func.function,
351 def _make_finalize_func(self, finalize_func): argument
354 finalize_func, self._transformation_name(),
453 def __init__(self, init_func, reduce_func, finalize_func): argument
456 self._finalize_func = finalize_func
467 def finalize_func(self): member in Reducer
Dprefetching_ops.py210 finalize_func=self._finalize_func,
/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.pbtxt1069 …nc_other_arguments\', \'key_func\', \'init_func\', \'reduce_func\', \'finalize_func\', \'output_ty…
1389 …args\', \'finalize_func_other_args\', \'init_func\', \'next_func\', \'finalize_func\', \'output_ty…
/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.pbtxt1069 …nc_other_arguments\', \'key_func\', \'init_func\', \'reduce_func\', \'finalize_func\', \'output_ty…
1389 …args\', \'finalize_func_other_args\', \'init_func\', \'next_func\', \'finalize_func\', \'output_ty…
/external/tensorflow/tensorflow/contrib/data/python/ops/
Dgrouping.py156 def __init__(self, init_func, reduce_func, finalize_func): argument
157 super(Reducer, self).__init__(init_func, reduce_func, finalize_func)
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dgroup_by_reducer_dataset_op.cc115 TF_RETURN_IF_ERROR(b->AddFunction(ctx, finalize_func().name())); in AsGraphDefInternal()
149 AttrValue finalize_func; in AsGraphDefInternal() local
150 b->BuildAttrValue(this->finalize_func(), &finalize_func); in AsGraphDefInternal()
174 {"finalize_func", finalize_func}, in AsGraphDefInternal()
404 const NameAttrList& finalize_func() const { in finalize_func() function in tensorflow::data::__anon898653860111::GroupByReducerDatasetOp::Dataset
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ExperimentalGroupByReducerDataset.pbtxt56 building a closure for `finalize_func`.
60 name: "finalize_func"
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dgroup_by_reducer_serialization_test.py35 finalize_func=lambda x: x)
/external/tensorflow/tensorflow/python/data/ops/
Dmulti_device_iterator_ops.py134 finalize_func=self._finalize_func,
177 finalize_func=self._finalize_func,
Ddataset_ops.py2459 def __init__(self, init_args, init_func, next_func, finalize_func): argument
2489 finalize_func,
2499 finalize_func=self._finalize_func.function,
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Doptimize_dataset_test.py72 finalize_func=lambda _: var)
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt11250 name: "finalize_func"
14342 name: "finalize_func"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt23365 name: "finalize_func"
28582 name: "finalize_func"
Dops_history.v1.pbtxt23736 name: "finalize_func"
28953 name: "finalize_func"