Home
last modified time | relevance | path

Searched refs:_flat_tensor_specs (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dtype_utils.py79 assert len(spec._flat_tensor_specs) == 1 # pylint: disable=protected-access
87 for t in spec._flat_tensor_specs # pylint: disable=protected-access
123 element_spec.__class__._flat_tensor_specs is # pylint: disable=protected-access
124 type_spec.BatchableTypeSpec._flat_tensor_specs): # pylint: disable=protected-access
H A Dtype_spec.py426 def _flat_tensor_specs(self): member in TypeSpec
515 expected = self._flat_tensor_specs
734 return spec._flat_tensor_specs # pylint: disable=protected-access
780 def _flat_tensor_specs(self) -> List[TypeSpec]: member in BatchableTypeSpec
843 return spec._flat_tensor_specs # pylint: disable=protected-access
H A Dtype_spec_test.py624 self.assertEqual(spec._flat_tensor_specs,
635 self.assertEqual(len(spec._flat_tensor_specs), len(full_type_list))
789 flat_specs = a_spec._flat_tensor_specs
809 self.assertEqual(len(a_spec._flat_tensor_specs), len(full_type_list))
H A Dsparse_tensor_test.py308 self.assertEqual(st_spec._flat_tensor_specs,
319 self.assertEqual(len(st_spec._flat_tensor_specs), len(full_type_list))
H A Dtensor_spec_test.py223 self.assertEqual(spec._flat_tensor_specs, [spec])
229 self.assertEqual(len(spec._flat_tensor_specs), len(full_type_list))
H A Dsparse_tensor.py349 def _flat_tensor_specs(self): member in SparseTensorSpec
H A Dtensor_spec.py233 def _flat_tensor_specs(self): member in TensorSpec
H A Dextension_type.py685 def _flat_tensor_specs(self): member in BatchableExtensionTypeSpec
H A Dextension_type_test.py1486 flat_specs = spec._flat_tensor_specs
1500 self.assertEqual(len(spec._flat_tensor_specs), len(full_type_list))
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dmap_fn.py552 tensor_specs.extend(spec._flat_tensor_specs) # pylint: disable=protected-access
576 tensor_list = elems_value_batchable[i:i + len(spec._flat_tensor_specs)]
608 num_tensors = len(spec._flat_tensor_specs)
/aosp_15_r20/external/tensorflow/tensorflow/python/data/util/
H A Dstructure.py214 flat_spec_lengths = [len(spec._flat_tensor_specs) for spec in flat_specs]
289 spec._flat_tensor_specs for spec in nest.flatten(element_spec)))
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_tensor_test.py2198 self.assertEqual(rt_spec._flat_tensor_specs,
2213 self.assertEqual(len(rt_spec._flat_tensor_specs), len(full_type_list))
H A Dragged_tensor.py2533 def _flat_tensor_specs(self): member in RaggedTensorSpec
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dinput_lib.py546 for component in spec._flat_tensor_specs: # pylint: disable=protected-access