Home
last modified time | relevance | path

Searched refs:output_structure (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/ops/
Dgrouping.py299 if not self._key_func.output_structure.is_compatible_with(
318 self._state_structure = self._init_func.output_structure
384 return self._finalize_func.output_structure
424 if not self._window_size_func.output_structure.is_compatible_with(
436 if not self._key_func.output_structure.is_compatible_with(
450 self._reduce_func.output_structure, dataset_ops.DatasetSpec):
454 self._reduce_func.output_structure._element_spec)
Dsnapshot.py232 if ((not self._shard_func.output_structure.is_compatible_with(
234 (not self._shard_func.output_structure.is_compatible_with(
Dtake_while_ops.py39 if not wrapped_func.output_structure.is_compatible_with(
Dbatching.py355 self._map_func.output_structure)
359 self._map_func.output_structure)
Doptimization.py192 …self._element_spec = self._funcs[0].output_structure._element_spec # pylint: disable=protected-ac…
Dprefetching_ops.py254 return self._map_func.output_structure
/external/tensorflow/tensorflow/python/framework/
Dop_def_library.py308 output_structure, is_stateful, op, outputs = _apply_op_helper(
310 if output_structure:
311 res = _Restructure(ops.convert_n_to_tensor(outputs), output_structure)
723 output_structure = []
727 output_structure.append(n)
730 output_structure.append(None)
733 output_structure.append(len(t.list.type))
735 output_structure.append(None)
765 return output_structure, op_def.is_stateful, op, outputs
/external/tensorflow/tensorflow/python/data/ops/
Diterator_ops.py204 output_structure = structure.convert_legacy_structure(
211 output_types=structure.get_flat_tensor_types(output_structure),
213 output_structure))
278 output_structure = structure.convert_legacy_structure(
283 output_types=structure.get_flat_tensor_types(output_structure),
284 output_shapes=structure.get_flat_tensor_shapes(output_structure))
Ddataset_ops.py3602 def output_structure(self): member in StructuredFunctionWrapper
3661 input_structure=self._init_func.output_structure)
3666 input_structure=self._init_func.output_structure)
4332 return self._map_func.output_structure
4382 return self._map_func.output_structure
4396 if not isinstance(self._map_func.output_structure, DatasetSpec):
4399 type(self._map_func.output_structure)))
4400 …self._structure = self._map_func.output_structure._element_spec # pylint: disable=protected-access
4428 if not isinstance(self._map_func.output_structure, DatasetSpec):
4431 type(self._map_func.output_structure)))
[all …]
Dreaders.py257 if not isinstance(self._map_func.output_structure, dataset_ops.DatasetSpec):
259 …self._element_spec = self._map_func.output_structure._element_spec # pylint: disable=protected-ac…
/external/tensorflow/tensorflow/python/data/benchmarks/
Dfrom_tensor_slices_benchmark.py41 …self._structure = self._map_func.output_structure._element_spec # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/keras/
Dbackend.py4073 output_structure = nest.pack_sequence_as(
4083 return nest.map_structure(self._eval_if_composite, output_structure)