Home
last modified time | relevance | path

Searched refs:_component_specs (Results 1 – 25 of 39) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dtype_spec.py355 def _component_specs(self): member in TypeSpec
423 self._component_specs, tensor_list, expand_composites=True))
428 return nest.flatten(self._component_specs, expand_composites=True)
784 self._component_specs)
794 self._component_specs,
805 get_spec_tensor_list, self._component_specs, self._to_components(value))
819 self._component_specs)
821 components = nest.map_structure_up_to(self._component_specs,
823 self._component_specs,
H A Dtype_utils.py129 element_spec._component_specs, # pylint: disable=protected-access
H A Dtype_spec_test.py68 def _component_specs(self): member in TwoTensorsSpec
135 def _component_specs(self): member in TwoCompositesSpec
183 _component_specs = property(lambda self: self.spec) variable in NestOfTensorsSpec
H A Dextension_type.py433 def _component_specs(self): # TypeSpec API. member in ExtensionTypeSpec
660 return spec._component_specs # pylint: disable=protected-access
H A Dindexed_slices.py254 def _component_specs(self): member in IndexedSlicesSpec
H A Dsparse_tensor.py325 def _component_specs(self): member in SparseTensorSpec
H A Dtensor_spec.py38 _component_specs = property(lambda self: self) variable in DenseSpec
H A Dcomposite_tensor_test.py44 _component_specs = property(lambda self: self.component_specs) variable in CTSpec
H A Dsparse_tensor_test.py265 self.assertEqual(st_spec._component_specs, expected)
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dnested_structure_coder_test.py484 _component_specs = property(lambda self: ()) variable in UnregisteredTypeSpec
494 _component_specs = property(lambda self: ()) variable in RegisteredTypeSpec
524 _component_specs = property(lambda self: ()) variable in UnregisteredTypeSpec
534 _component_specs = property(lambda self: ()) variable in RegisteredTypeSpec
H A Dnested_structure_coder.py527 …nest.flatten(type_spec_value._component_specs, expand_composites=True)) # pylint: disable=protect…
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dvariable_spec_test.py38 self.assertEqual(spec._component_specs,
47 self.assertEqual(spec2._component_specs,
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dinput_lib.py818 def _component_specs(self): member in DistributedIteratorSpec
962 def _component_specs(self): member in DistributedDatasetSpec
1281 def _component_specs(self): member in DistributedDatasetsFromFunctionSpec
1681 def _component_specs(self): member in _SingleWorkerDatasetIteratorSpec
H A Dsharded_variable.py244 def _component_specs(self): member in ShardedVariableSpec
H A Dinput_lib_test.py957 feature_specs = data.element_spec["feature"]._component_specs
958 value_specs = data.element_spec["label"]._component_specs
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Doptional_ops.py237 def _component_specs(self): member in OptionalSpec
H A Diterator_ops.py910 def _component_specs(self): member in IteratorSpec
H A Dmulti_device_iterator_ops.py385 def _component_specs(self): member in MultiDeviceIteratorSpec
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/
H A Dparameter_server_evaluation_test.py61 def _component_specs(self): member in KerasMetricTypeSpec
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/
H A Dvalues.py307 def _component_specs(self): member in PerWorkerValuesTypeSpec
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dkeras_tensor.py489 def _component_specs(self): member in UserRegisteredSpec
/aosp_15_r20/external/tensorflow/tensorflow/python/data/util/
H A Dstructure.py520 def _component_specs(self): member in NoneTensorSpec
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dnest.py304 … yield iterable.value_type.__name__, iterable._component_specs # pylint: disable=protected-access
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_tensor_supported_values_test.py400 self.assertEqual(rt_spec._component_specs, expected)
/aosp_15_r20/external/tensorflow/tensorflow/python/module/
H A Dmodule_test.py266 def _component_specs(self): member in VariableTrackingTest.test_composite_variable.Spec

12