Searched refs:_init_func (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | grouping.py | 285 self._init_func.function.captured_inputs,
289 init_func=self._init_func.function,
308 self._init_func = dataset_ops.StructuredFunctionWrapper(
318 self._state_structure = self._init_func.output_structure
319 state_types = self._init_func.output_types
320 state_shapes = self._init_func.output_shapes
321 state_classes = self._init_func.output_classes
348 (self._init_func.output_types, wrapped_func.output_types))
369 self._init_func.output_shapes, weakened_state_shapes)
388 self._key_func, self._init_func, self._reduce_func, self._finalize_func
[all …]
|
| D | prefetching_ops.py | 108 def _init_func(): function
121 …init_func_concrete = _init_func._get_concrete_function_internal() # pylint: disable=protected-acc…
131 …self._init_func = _remote_init_func._get_concrete_function_internal() # pylint: disable=protected…
132 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 | 51 def _init_func(): function
54 init_func_concrete = _init_func.get_concrete_function()
65 self._init_func = _remote_init_func.get_concrete_function()
66 self._init_captured_args = self._init_func.captured_inputs
136 init_func=self._init_func,
161 self._init_func = per_device_dataset._init_func
162 self._init_captured_args = self._init_func.captured_inputs
178 init_func=self._init_func,
|
| D | dataset_ops.py | 3653 self._init_func = StructuredFunctionWrapper(
3661 input_structure=self._init_func.output_structure)
3666 input_structure=self._init_func.output_structure)
3672 self._init_func.function.captured_inputs,
3675 init_func=self._init_func.function,
|