Searched refs:_flat_tensor_specs (Results 1 – 9 of 9) sorted by relevance
555 tensor_specs.extend(spec._flat_tensor_specs) # pylint: disable=protected-access579 tensor_list = elems_value_batchable[i:i + len(spec._flat_tensor_specs)]611 num_tensors = len(spec._flat_tensor_specs)
210 flat_spec_lengths = [len(spec._flat_tensor_specs) for spec in flat_specs]283 return functools.reduce(lambda state, value: state + value._flat_tensor_specs,
274 def _flat_tensor_specs(self): member in TypeSpec339 expected = self._flat_tensor_specs
352 def _flat_tensor_specs(self): member in SparseTensorSpec
310 self.assertEqual(st_spec._flat_tensor_specs,
441 self.assertEqual(spec._flat_tensor_specs,
1101 def _flat_tensor_specs(self): member in StructuredTensorSpec1105 result.extend(field_spec._flat_tensor_specs)1120 num_tensors_for_field = len(field_spec._flat_tensor_specs)
2395 def _flat_tensor_specs(self): member in RaggedTensorSpec
1834 self.assertEqual(rt_spec._flat_tensor_specs,