Searched defs:batch_size_tensor (Results 1 – 3 of 3) sorted by relevance
152 Tensor batch_size_tensor; in MutateBatchSize() local
85 self, batch_size_tensor, static_batch_size=None): argument
2755 def _generate_zero_filled_state(batch_size_tensor, state_size, dtype): argument