Searched refs:_init_func (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | grouping.py | 259 self._init_func.function.captured_inputs,
263 init_func=self._init_func.function,
282 self._init_func = dataset_ops.StructuredFunctionWrapper(
292 self._state_structure = self._init_func.output_structure
293 state_types = self._init_func.output_types
294 state_shapes = self._init_func.output_shapes
295 state_classes = self._init_func.output_classes
323 (self._init_func.output_types, wrapped_func.output_types))
344 self._init_func.output_shapes, weakened_state_shapes)
363 self._key_func, self._init_func, self._reduce_func, self._finalize_func
[all …]
|
| D | prefetching_ops.py | 107 def _init_func(): function
120 …init_func_concrete = _init_func._get_concrete_function_internal() # pylint: disable=protected-acc…
130 …self._init_func = _remote_init_func._get_concrete_function_internal() # pylint: disable=protected…
131 self._init_captured_args = self._init_func.captured_inputs
198 self._init_func.add_to_graph(g)
208 init_func=self._init_func,
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 47 def _init_func(): function
50 …init_func_concrete = _init_func._get_concrete_function_internal() # pylint: disable=protected-acc…
61 …self._init_func = _remote_init_func._get_concrete_function_internal() # pylint: disable=protected…
62 self._init_captured_args = self._init_func.captured_inputs
132 init_func=self._init_func,
158 self._init_func = per_device_dataset._init_func
159 self._init_captured_args = self._init_func.captured_inputs
175 init_func=self._init_func,
|
| D | dataset_ops.py | 2478 self._init_func = StructuredFunctionWrapper(
2486 input_structure=self._init_func.output_structure)
2491 input_structure=self._init_func.output_structure)
2494 + self._init_func.function.captured_inputs,
2497 init_func=self._init_func.function,
|