Searched defs:_unbatch (Results 1 – 10 of 10) sorted by relevance
402 def _unbatch(self): member in SparseTensorSpec
227 def _unbatch(self): member in TensorSpec
159 def _unbatch(self): member in TwoCompositesSpec
770 def _unbatch(self) -> TypeSpec: member in BatchableTypeSpec
672 def _unbatch(self): member in BatchableExtensionTypeSpec
539 def _unbatch(self): member in NoneTensorSpec
1329 def _unbatch(self, batch_size): member in _LinearOperatorSpec
2610 def _unbatch(self): member in RaggedTensorSpec
2508 def _unbatch(self): member in ParticleSpec
4593 def _unbatch(self): member in DatasetSpec