| /external/tensorflow/tensorflow/python/data/ops/ |
| D | optional_ops.py | 105 def element_spec(self): member in Optional
119 def empty(element_spec): argument
137 return _OptionalImpl(gen_dataset_ops.optional_none(), element_spec)
158 element_spec = structure.type_spec_from_value(value)
159 encoded_value = structure.to_tensor_list(element_spec, value)
163 element_spec)
173 def __init__(self, variant_tensor, element_spec): argument
175 self._element_spec = element_spec
197 def element_spec(self): member in _OptionalImpl
231 def __init__(self, element_spec): argument
[all …]
|
| D | iterator_ops.py | 326 dataset.element_spec)
329 dataset.element_spec)
332 dataset.element_spec)
429 output_types=structure.get_flat_tensor_types(self.element_spec),
431 self.element_spec)), self.element_spec)
493 def element_spec(self): member in Iterator
582 def element_spec(self): member in IteratorBase
646 def __init__(self, dataset=None, components=None, element_spec=None): argument
670 if (components is None or element_spec is None):
673 self._element_spec = element_spec
[all …]
|
| D | multi_device_iterator_ops.py | 42 source_device, element_spec): argument
43 self._element_spec = element_spec
147 def element_spec(self): member in _PerDeviceGenerator
160 self._element_spec = per_device_dataset.element_spec
189 def element_spec(self): member in _ReincarnatedPerDeviceGenerator
277 self._dataset.element_spec)
368 def element_spec(self): member in MultiDeviceIterator
369 return self._dataset.element_spec
415 def __init__(self, devices, source_device, element_spec): argument
418 self._element_spec = element_spec
[all …]
|
| D | dataset_ops.py | 459 def element_spec(self): member in DatasetV2
528 for component_spec in nest.flatten(self.element_spec):
545 return structure.get_flat_tensor_shapes(self.element_spec)
554 return structure.get_flat_tensor_types(self.element_spec)
577 return DatasetSpec(self.element_spec)
2146 input_structure=(state_structure, self.element_spec),
2515 self.element_spec)
2529 self.element_spec)
2543 self.element_spec)
2546 def element_spec(self): member in DatasetV1
[all …]
|
| D | readers.py | 138 def element_spec(self): member in _TextLineDataset
187 def element_spec(self): member in TextLineDatasetV2
244 def element_spec(self): member in _TFRecordDataset
295 def element_spec(self): member in ParallelInterleaveDataset
364 def element_spec(self): member in TFRecordDatasetV2
449 def element_spec(self): member in _FixedLengthRecordDataset
507 def element_spec(self): member in FixedLengthRecordDatasetV2
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure.py | 189 def _from_tensor_list_helper(decode_fn, element_spec, tensor_list): argument
209 flat_specs = nest.flatten(element_spec)
221 return nest.pack_sequence_as(element_spec, flat_ret)
224 def from_compatible_tensor_list(element_spec, tensor_list): argument
244 element_spec, tensor_list)
247 def from_tensor_list(element_spec, tensor_list): argument
267 lambda spec, value: spec._from_tensor_list(value), element_spec,
271 def get_flat_tensor_specs(element_spec): argument
284 nest.flatten(element_spec), [])
287 def get_flat_tensor_shapes(element_spec): argument
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | io.py | 97 input_structure=dataset.element_spec,
101 encoded = coder.encode_structure(dataset.element_spec)
124 def __init__(self, path, element_spec=None, compression=None, argument
134 if element_spec is None:
143 self._element_spec = element_spec
164 def element_spec(self): member in _LoadDataset
169 def load(path, element_spec=None, compression=None, reader_func=None): argument
230 element_spec=element_spec,
|
| D | data_service_ops.py | 179 def element_spec(self): member in _DataServiceDatasetV2
241 element_spec, argument
301 if element_spec is None:
317 lambda x: compression_ops.uncompress(x, output_spec=element_spec),
384 dataset.element_spec,
686 element_spec=None, argument
772 element_spec=element_spec,
|
| D | distribute.py | 72 self._element_spec = input_dataset.element_spec
84 def element_spec(self): member in _AutoShardDataset
222 def element_spec(self): member in _RebatchDataset
293 def element_spec(self): member in _LegacyRebatchDataset
300 def __init__(self, graph_def, device, element_spec): argument
301 self._elem_spec = element_spec
307 def element_spec(self): member in _RemoteDataset
348 ds = _RemoteDataset(graph_def, device, dataset.element_spec)
|
| D | optimization.py | 100 self._element_spec = self._datasets[0].element_spec
112 def element_spec(self): member in _ChooseFastestDataset
186 input_structure = dataset_ops.DatasetSpec(input_dataset.element_spec)
222 def element_spec(self): member in _ChooseFastestBranchDataset
|
| D | batching.py | 323 def element_spec(self): member in _DenseToSparseBatchDataset
376 def element_spec(self): member in _MapAndBatchDataset
420 input_dataset.element_spec)
440 …if dataset_ops._should_unpack_args(input_dataset.element_spec): # pylint: disable=protected-access
451 def element_spec(self): member in _DenseToRaggedDataset
|
| D | compression_ops.py | 34 element_spec = structure.type_spec_from_value(element)
35 tensor_list = structure.to_tensor_list(element_spec, element)
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | optional_test.py | 165 self.assertTrue(opt.element_spec.is_compatible_with(value_structure))
167 opt.element_spec.is_compatible_with(
170 opt.element_spec.is_compatible_with(
189 add_opt = optional_ops._OptionalImpl(add_tensor, opt1.element_spec)
193 opt_none1 = optional_ops.Optional.empty(opt1.element_spec)
194 opt_none2 = optional_ops.Optional.empty(opt2.element_spec)
197 add_opt = optional_ops._OptionalImpl(add_tensor, opt_none1.element_spec)
214 add_opt = optional_ops._OptionalImpl(add_tensor, opt3.element_spec)
218 opt1.element_spec)
231 zeros_opt = optional_ops._OptionalImpl(zeros_tensor, opt.element_spec)
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 220 def element_spec(self): member in DistributedIteratorInterface
459 def element_spec(self): member in DistributedDatasetInterface
582 def _is_statically_shaped(element_spec): argument
595 for spec in nest.flatten(element_spec):
750 def element_spec(self): member in DistributedIteratorV1
763 def __init__(self, input_workers, element_spec, strategy, argument
773 self._element_spec = element_spec
813 element_spec = nest.map_structure(
816 return DistributedIteratorSpec(self._input_workers, element_spec,
827 element_spec = nest.map_structure(
[all …]
|
| D | input_lib_type_spec_test.py | 188 element_spec = ds.element_spec
189 iter_element_spec = iter(ds).element_spec
190 nest.assert_same_structure(element_spec, iter_element_spec)
192 nest.flatten(element_spec), nest.flatten(iter_element_spec))
194 @def_function.function(input_signature=[element_spec])
219 @def_function.function(input_signature=[dist_dataset.element_spec])
315 element_spec = ds.element_spec
316 iter_element_spec = iter(ds).element_spec
317 nest.assert_same_structure(element_spec, iter_element_spec)
319 nest.flatten(element_spec), nest.flatten(iter_element_spec))
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | io_test.py | 53 self._test_dir, dataset.element_spec, compression=compression)
60 dataset2 = io.load(self._test_dir, dataset.element_spec)
67 dataset2 = io.load(self._test_dir, dataset.element_spec)
79 dataset.element_spec,
99 self._test_dir, dataset.element_spec, compression=compression)
|
| D | compression_ops_test.py | 108 element_spec = dataset.element_spec
111 dataset = dataset.map(lambda x: compression_ops.uncompress(x, element_spec))
|
| D | wrap_unwrap_test.py | 42 ds.element_spec)
59 ds.element_spec)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.experimental.-optional.pbtxt | 7 name: "element_spec"
15 argspec: "args=[\'element_spec\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.data.experimental.-optional.pbtxt | 7 name: "element_spec"
15 argspec: "args=[\'element_spec\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.data.-dataset-spec.pbtxt | 8 name: "element_spec"
17 …argspec: "args=[\'self\', \'element_spec\', \'dataset_shape\'], varargs=None, keywords=None, defau…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.experimental.-optional.pbtxt | 7 name: "element_spec"
15 argspec: "args=[\'element_spec\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.data.experimental.-optional.pbtxt | 7 name: "element_spec"
15 argspec: "args=[\'element_spec\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.data.experimental.-dataset-structure.pbtxt | 8 name: "element_spec"
17 …argspec: "args=[\'self\', \'element_spec\', \'dataset_shape\'], varargs=None, keywords=None, defau…
|
| D | tensorflow.data.-dataset-spec.pbtxt | 8 name: "element_spec"
17 …argspec: "args=[\'self\', \'element_spec\', \'dataset_shape\'], varargs=None, keywords=None, defau…
|