Home
last modified time | relevance | path

Searched defs:element_spec (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/python/data/util/
Dstructure.py189 def _from_tensor_list_helper(decode_fn, element_spec, tensor_list): argument
224 def from_compatible_tensor_list(element_spec, tensor_list): argument
247 def from_tensor_list(element_spec, tensor_list): argument
271 def get_flat_tensor_specs(element_spec): argument
287 def get_flat_tensor_shapes(element_spec): argument
300 def get_flat_tensor_types(element_spec): argument
313 def _to_tensor_list_helper(encode_fn, element_spec, element): argument
343 def to_batched_tensor_list(element_spec, element): argument
369 def to_tensor_list(element_spec, element): argument
/external/tensorflow/tensorflow/python/data/ops/
Doptional_ops.py105 def element_spec(self): member in Optional
119 def empty(element_spec): argument
173 def __init__(self, variant_tensor, element_spec): argument
197 def element_spec(self): member in _OptionalImpl
231 def __init__(self, element_spec): argument
Dmulti_device_iterator_ops.py43 source_device, element_spec): argument
148 def element_spec(self): member in _PerDeviceGenerator
190 def element_spec(self): member in _ReincarnatedPerDeviceGenerator
357 def element_spec(self): member in MultiDeviceIterator
404 def __init__(self, devices, source_device, element_spec): argument
466 element_spec=None): argument
597 def element_spec(self): member in OwnedMultiDeviceIterator
Dreaders.py138 def element_spec(self): member in _TextLineDataset
219 def element_spec(self): member in TextLineDatasetV2
276 def element_spec(self): member in _TFRecordDataset
327 def element_spec(self): member in ParallelInterleaveDataset
437 def element_spec(self): member in TFRecordDatasetV2
522 def element_spec(self): member in _FixedLengthRecordDataset
611 def element_spec(self): member in FixedLengthRecordDatasetV2
Ddataset_ops.py453 def element_spec(self): member in DatasetV2
3288 def element_spec(self): member in DatasetV1
3547 def element_spec(self): member in DatasetV1Adapter
3775 def element_spec(self): member in UnaryUnchangedStructureDataset
3794 def element_spec(self): member in TensorDataset
3821 def element_spec(self): member in TensorSliceDataset
3850 def element_spec(self): member in SparseTensorSliceDataset
3865 def element_spec(self): member in _VariantDataset
3871 def __init__(self, variant_tensor, element_spec, dataset_shape): argument
3924 def __init__(self, element_spec, dataset_shape=()): argument
[all …]
Diterator_ops.py511 def element_spec(self): member in Iterator
610 def element_spec(self): member in IteratorBase
677 def __init__(self, dataset=None, components=None, element_spec=None): argument
836 def element_spec(self): member in OwnedIterator
895 def __init__(self, element_spec): argument
/external/tensorflow/tensorflow/python/data/experimental/ops/
Ddistribute.py89 def element_spec(self): member in _AutoShardDataset
227 def element_spec(self): member in _RebatchDataset
298 def element_spec(self): member in _LegacyRebatchDataset
305 def __init__(self, graph_def, device, element_spec): argument
312 def element_spec(self): member in _RemoteDataset
Dio.py203 def __init__(self, path, element_spec=None, compression=None, argument
243 def element_spec(self): member in _LoadDataset
248 def load(path, element_spec=None, compression=None, reader_func=None): argument
Ddata_service_ops.py183 element_spec, argument
304 def element_spec(self): member in _DataServiceDatasetV2
312 def __init__(self, dataset_id, processing_mode, address, element_spec, argument
807 element_spec, argument
958 element_spec=None, argument
Dbatching.py323 def element_spec(self): member in _DenseToSparseBatchDataset
376 def element_spec(self): member in _MapAndBatchDataset
451 def element_spec(self): member in _DenseToRaggedDataset
Dmatching_files.py38 def element_spec(self): member in MatchingFilesDataset
Dparsing_ops.py104 def element_spec(self): member in _ParseExampleDataset
Dprefetching_ops.py258 def element_spec(self): member in _MapOnGpuDataset
Dinterleave_ops.py149 def element_spec(self): member in _DirectedInterleaveDataset
Dgrouping.py374 def element_spec(self): member in _GroupByReducerDataset
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib.py276 def element_spec(self): member in DistributedIteratorInterface
515 def element_spec(self): member in DistributedDatasetInterface
658 def _is_statically_shaped(element_spec): argument
833 def element_spec(self): member in DistributedIteratorV1
849 element_spec, argument
906 def __init__(self, input_workers, element_spec, strategy, argument
991 element_spec=None, argument
1020 def element_spec(self): member in DistributedIterator
1079 def __init__(self, input_workers, element_spec, strategy, argument
1152 element_spec=None, argument
[all …]
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dvalues.py373 def element_spec(self): member in PerWorkerDatasetFromDatasetFunction
397 def __init__(self, graph_def, element_spec): argument
403 def element_spec(self): member in _RemoteDataset
407 def deserialize_dataset_from_graph(graph_def, element_spec): argument
/external/tensorflow/tensorflow/python/data/benchmarks/
Dfrom_tensor_slices_benchmark.py54 def element_spec(self): member in SingleThreadedFlatMapDataset
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddata_service_ops.cc48 tstring element_spec; in RegisterDatasetOp() local
/external/tensorflow/tensorflow/core/data/service/
Ddispatcher_client.cc130 const DatasetDef& dataset, const absl::optional<std::string>& element_spec, in RegisterDataset()
240 std::string& element_spec) { in GetElementSpec()
Ddispatcher.proto75 bytes element_spec = 2; field
93 bytes element_spec = 1; field
Dworker_client_test.cc69 absl::optional<std::string> element_spec; in RegisterDataset() local
Ddispatcher_state_test.cc59 Status SetElementSpec(int64_t dataset_id, const std::string& element_spec, in SetElementSpec()
148 std::string element_spec = "test_element_spec"; in TEST() local
157 std::string element_spec; in TEST() local
/external/tensorflow/tensorflow/python/data/experimental/service/
Dserver_lib_wrapper.cc103 std::string element_spec; in PYBIND11_MODULE() local
/external/tensorflow/tensorflow/python/keras/distribute/
Dparameter_server_evaluation_test.py92 def element_spec(self): member in MeanMetricAsCompositeTensor

12