Searched refs:batch_size_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | rebatch.cc | 152 Tensor batch_size_tensor; in MutateBatchSize() local
154 GetNodeAttr(*batch_size_node, "value", &batch_size_tensor)); in MutateBatchSize()
155 if (!TensorShapeUtils::IsScalar(batch_size_tensor.shape())) { in MutateBatchSize()
158 int64 batch_size = batch_size_tensor.scalar<int64>()(); in MutateBatchSize()
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | estimators.py | 85 self, batch_size_tensor, static_batch_size=None): argument
108 [batch_size_tensor[None],
206 batch_size_tensor=array_ops.shape(
279 batch_size_tensor = array_ops.shape(time_placeholder)[0]
282 batch_size_tensor, static_batch_size=default_batch_size))
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| D | beam_search_decoder_test.py | 496 batch_size_tensor = constant_op.constant(batch_size)
523 dtype=dtypes.float32, batch_size=batch_size_tensor * beam_width)
529 start_tokens=array_ops.fill([batch_size_tensor], start_token),
610 batch_size_tensor = constant_op.constant(batch_size)
637 dtype=dtypes.float32, batch_size=batch_size_tensor * beam_width)
651 start_tokens=array_ops.fill([batch_size_tensor], start_token),
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent.py | 2755 def _generate_zero_filled_state(batch_size_tensor, state_size, dtype): argument
2757 if batch_size_tensor is None or dtype is None:
2760 'batch_size={}, dtype={}'.format(batch_size_tensor, dtype))
2764 init_state_size = [batch_size_tensor] + flat_dims
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_estimator.py | 3634 batch_size_tensor = constant_op.constant(batch_size, dtypes.int32)
3637 batch_size_tensor,
3639 data=(batch_size_tensor, real_batch_size),
3643 missing_count = batch_size_tensor - real_batch_size
|