Searched refs:_to_batched_tensor_list (Results 1 – 10 of 10) sorted by relevance
365 lambda state, spec, component: state + spec._to_batched_tensor_list(529 def _to_batched_tensor_list(self, value): member in NoneTensorSpec
230 return spec._to_batched_tensor_list(value)417 for t in spec._to_batched_tensor_list(x)]
257 batched_tensor_list = batched_spec._to_batched_tensor_list(batched)
1111 def _to_batched_tensor_list(self, value): member in StructuredTensorSpec1146 result.extend(field_spec._to_batched_tensor_list(field_value))
568 elems_batchable.extend(spec._to_batched_tensor_list(elems_tensor))
365 def _to_batched_tensor_list(self, value): member in SparseTensorSpec
526 def _to_batched_tensor_list(self, value): member in BatchableTypeSpec
1875 tensor_list = rt_spec._to_batched_tensor_list(rt)1886 tensor_list = rt_spec._to_batched_tensor_list(rt)
2419 def _to_batched_tensor_list(self, value): member in RaggedTensorSpec
3382 def _to_batched_tensor_list(self, value): member in DatasetSpec