Searched refs:unbatched (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor_spec_test.py | 174 def testBatchUnbatch(self, unbatched, batch_size, batched): argument
175 self.assertEqual(unbatched._batch(batch_size), batched)
176 self.assertEqual(batched._unbatch(), unbatched)
235 def testBatchUnbatchValues(self, unbatched, batch_size, batched, argument
238 unbatched = unbatched() # Deferred init because it creates tensors.
244 unbatched_spec = type_spec.type_spec_from_value(unbatched[0])
246 for st in unbatched]
263 self.assertLen(actual_unbatched, len(unbatched))
266 for (actual, expected) in zip(actual_unbatched, unbatched):
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_RaggedTensorToVariant.pbtxt | 35 `batched_input` is True, then input `RaggedTensor` is unbatched along the
|