Searched refs:_finalize_func (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 108 def _finalize_func(unused_string_handle): function
111 …finalize_func_concrete = _finalize_func._get_concrete_function_internal() # pylint: disable=prote…
124 …self._finalize_func = _remote_finalize_func._get_concrete_function_internal( # pylint: disable=pr…
126 self._finalize_captured_args = self._finalize_func.captured_inputs
134 finalize_func=self._finalize_func,
168 self._finalize_func = per_device_dataset._finalize_func
177 finalize_func=self._finalize_func,
|
| D | dataset_ops.py | 2488 self._finalize_func = StructuredFunctionWrapper(
2496 self._finalize_func.function.captured_inputs,
2499 finalize_func=self._finalize_func.function,
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | prefetching_ops.py | 167 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,
|
| D | grouping.py | 261 self._finalize_func.function.captured_inputs,
265 finalize_func=self._finalize_func.function,
353 self._finalize_func = dataset_ops.StructuredFunctionWrapper(
359 return self._finalize_func.output_structure
363 self._key_func, self._init_func, self._reduce_func, self._finalize_func
456 self._finalize_func = finalize_func
468 return self._finalize_func
|