Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/python/data/util/
H A Dstructure.py193 def _from_tensor_list_helper(decode_fn, element_spec, tensor_list): argument
228 def from_compatible_tensor_list(element_spec, tensor_list): argument
251 def from_tensor_list(element_spec, tensor_list): argument
275 def get_flat_tensor_specs(element_spec): argument
292 def get_flat_tensor_shapes(element_spec): argument
305 def get_flat_tensor_types(element_spec): argument
318 def _to_tensor_list_helper(encode_fn, element_spec, element): argument
348 def to_batched_tensor_list(element_spec, element): argument
374 def to_tensor_list(element_spec, element): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Ddataset_ops.py535 def element_spec(self): member in DatasetV2
1766 def load(path, element_spec=None, compression=None, reader_func=None): argument
3864 def element_spec(self): member in DatasetV1
4161 def element_spec(self): member in DatasetV1Adapter
4405 def element_spec(self): member in UnaryUnchangedStructureDataset
4412 def __init__(self, dataset_variant, element_spec): argument
4420 def element_spec(self): member in _VariantDataset
4426 def __init__(self, variant_tensor, element_spec, dataset_shape): argument
4479 def __init__(self, element_spec, dataset_shape=()): argument
4488 def element_spec(self): member in DatasetSpec
[all …]
H A Dmulti_device_iterator_ops.py40 source_device, element_spec, iterator_is_anonymous): argument
161 def element_spec(self): member in _PerDeviceGenerator
203 def element_spec(self): member in _ReincarnatedPerDeviceGenerator
363 def element_spec(self): member in MultiDeviceIterator
372 def __init__(self, devices, source_device, element_spec): argument
433 element_spec=None): argument
566 def element_spec(self): member in OwnedMultiDeviceIterator
H A Doptional_ops.py98 def element_spec(self): member in Optional
112 def empty(element_spec): argument
166 def __init__(self, variant_tensor, element_spec): argument
191 def element_spec(self): member in _OptionalImpl
225 def __init__(self, element_spec): argument
H A Dreaders.py163 def element_spec(self): member in _TextLineDataset
248 def element_spec(self): member in TextLineDatasetV2
313 def element_spec(self): member in _TFRecordDataset
375 def element_spec(self): member in ParallelInterleaveDataset
478 def element_spec(self): member in TFRecordDatasetV2
561 def element_spec(self): member in _FixedLengthRecordDataset
657 def element_spec(self): member in FixedLengthRecordDatasetV2
H A Diterator_ops.py523 def element_spec(self): member in Iterator
598 def element_spec(self): member in IteratorBase
665 def __init__(self, dataset=None, components=None, element_spec=None): argument
836 def element_spec(self): member in OwnedIterator
899 def __init__(self, element_spec): argument
H A Dload_op.py33 element_spec, argument
42 def __init__(self, path, element_spec=None, compression=None, argument
84 def element_spec(self): member in _LoadDataset
H A Dsave_op.py96 def element_spec(self): member in _SaveDataset
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dinput_lib.py169 def element_spec(self): member in DistributedIteratorInterface
408 def element_spec(self): member in DistributedDatasetInterface
523 def _is_statically_shaped(element_spec): argument
676 element_spec, argument
866 element_spec=None, argument
897 def element_spec(self): member in DistributedIterator
1006 element_spec=None, argument
1251 def element_spec(self): member in DistributedDataset
1324 element_spec=None, argument
1454 def element_spec(self): member in DistributedDatasetsFromFunction
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Ddistribute.py85 def element_spec(self): member in _AutoShardDataset
236 def element_spec(self): member in _RebatchDataset
313 def element_spec(self): member in _LegacyRebatchDataset
320 def __init__(self, graph_def, device, element_spec): argument
327 def element_spec(self): member in _RemoteDataset
H A Ddata_service_ops.py243 element_spec, argument
394 def element_spec(self): member in _DataServiceDatasetV2
402 def __init__(self, dataset_id, processing_mode, address, element_spec, argument
928 element_spec, argument
1087 element_spec=None, argument
H A Dbatching.py328 def element_spec(self): member in _DenseToSparseBatchDataset
381 def element_spec(self): member in _MapAndBatchDataset
456 def element_spec(self): member in _DenseToRaggedDataset
H A Dmatching_files.py34 def element_spec(self): member in MatchingFilesDataset
H A Dfrom_list.py71 def element_spec(self): member in _ListDataset
H A Dreaders.py820 def element_spec(self): member in CsvDatasetV2
1198 def element_spec(self): member in SqlDatasetV2
H A Dio.py111 def load(path, element_spec=None, compression=None, reader_func=None): argument
H A Dparsing_ops.py101 def element_spec(self): member in _ParseExampleDataset
H A Dlookup_ops.py27 def _check_table_initializer_element_spec(element_spec): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dtype_utils.py92 def _specs_for_flat_tensors(element_spec): argument
142 def fulltypes_for_flat_tensors(element_spec): argument
174 def iterator_full_type_from_spec(element_spec): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/
H A Dvalues.py370 def element_spec(self): member in PerWorkerDatasetFromDatasetFunction
394 def __init__(self, graph_def, element_spec): argument
400 def element_spec(self): member in _RemoteDataset
404 def deserialize_dataset_from_graph(graph_def, element_spec): argument
/aosp_15_r20/external/federated-compute/fcp/tensorflow/
H A Dexternal_dataset.py50 def element_spec(self): member in ExternalDataset
/aosp_15_r20/external/tensorflow/tensorflow/python/data/benchmarks/
H A Dfrom_tensor_slices_benchmark.py51 def element_spec(self): member in SingleThreadedFlatMapDataset
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Ddata_service.pb.h843 inline const std::string& DataServiceMetadata::element_spec() const { in element_spec() function
893 inline void DataServiceMetadata::set_allocated_element_spec(std::string* element_spec) { in set_allocated_element_spec()
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Ddata_service.proto68 bytes element_spec = 1; field
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Ddata_service_ops.cc69 tstring element_spec; in RegisterDatasetOp() local

12