Home
last modified time | relevance | path

Searched refs:static_batch_size (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/contrib/training/python/training/
Dbucket_ops.py187 static_batch_size = tensor_util.constant_value(bucket_batch_size)
188 batch_size[i] = (static_batch_size if static_batch_size is not None else
212 else static_batch_size)
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Dbeam_search_decoder.py388 static_batch_size = tensor_util.constant_value(self._batch_size)
391 if static_batch_size is None else static_batch_size * self._beam_width)
427 static_batch_size = tensor_util.constant_value(self._batch_size)
429 [static_batch_size, self._beam_width]).concatenate(s)
967 static_batch_size = tensor_util.constant_value(batch_size)
1022 next_beam_scores.set_shape([static_batch_size, beam_width])
1023 word_indices.set_shape([static_batch_size, beam_width])
1378 static_batch_size = tensor_util.constant_value(batch_size)
1380 tensor_shape.TensorShape([static_batch_size]).concatenate(
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Destimators.py85 self, batch_size_tensor, static_batch_size=None): argument
104 (static_batch_size,)).concatenate(state.shape)
208 static_batch_size=default_batch_size))
282 batch_size_tensor, static_batch_size=default_batch_size))
Dar_model.py440 static_batch_size = times.get_shape().dims[0].value
442 [static_batch_size, self.input_window_size, input_feature_size])
444 [static_batch_size, self.output_window_size, output_feature_size])
/external/tensorflow/tensorflow/contrib/gan/python/features/python/
Dvirtual_batchnorm_impl.py44 static_batch_size = tensor_util.constant_value(batch_size)
45 return static_batch_size or batch_size
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining.py1773 static_batch_size = training_utils.get_static_batch_size(first_layer)
1774 if static_batch_size is not None:
1777 if batch_size is not None and batch_size != static_batch_size:
1780 '{}'.format(batch_size, static_batch_size))
1787 if ds_batch_size is not None and ds_batch_size != static_batch_size:
1791 ds_batch_size, static_batch_size))
1795 batch_size = static_batch_size
/external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/ops/
Dseq2seq.py526 static_batch_size = encoder_inputs[0].get_shape()[0]
528 static_batch_size.merge_with(inp.get_shape()[0])
529 batch_size = static_batch_size.value
/external/tensorflow/tensorflow/python/ops/
Drnn.py1097 static_batch_size = tensor_shape.dimension_at_index(input_shape[0], 0)
1101 static_batch_size.merge_with(
1104 batch_size = tensor_shape.dimension_value(static_batch_size)
Drnn_cell_impl.py288 static_batch_size = tensor_util.constant_value(
291 static_batch_size = batch_size
292 if inputs.shape.dims[0].value != static_batch_size: