Home
last modified time | relevance | path

Searched refs:_finalize_func (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/ops/
Dprefetching_ops.py167 def _finalize_func(string_handle): function
183 …finalize_func_concrete = _finalize_func._get_concrete_function_internal() # pylint: disable=prote…
193 …self._finalize_func = _remote_finalize_func._get_concrete_function_internal( # pylint: disable=pr…
195 self._finalize_captured_args = self._finalize_func.captured_inputs
200 self._finalize_func.add_to_graph(g)
210 finalize_func=self._finalize_func,
Dgrouping.py287 self._finalize_func.function.captured_inputs,
291 finalize_func=self._finalize_func.function,
378 self._finalize_func = dataset_ops.StructuredFunctionWrapper(
384 return self._finalize_func.output_structure
388 self._key_func, self._init_func, self._reduce_func, self._finalize_func
480 self._finalize_func = finalize_func
492 return self._finalize_func
/external/tensorflow/tensorflow/python/data/ops/
Dmulti_device_iterator_ops.py113 def _finalize_func(unused_string_handle): function
116 finalize_func_concrete = _finalize_func.get_concrete_function()
129 self._finalize_func = _remote_finalize_func.get_concrete_function()
130 self._finalize_captured_args = self._finalize_func.captured_inputs
138 finalize_func=self._finalize_func,
171 self._finalize_func = per_device_dataset._finalize_func
180 finalize_func=self._finalize_func,
Ddataset_ops.py3663 self._finalize_func = StructuredFunctionWrapper(
3674 self._finalize_func.function.captured_inputs,
3677 finalize_func=self._finalize_func.function,