| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| mpi_allreduce_test.py | 60 hidden_size = batch_size
64 inputs = tf.placeholder(tf.float32, shape=(batch_size, hidden_size),
72 shape=(hidden_size, out_size),
91 my_reducer = tf.Variable(initial_value=np.ones((hidden_size, out_size)),
112 input_feed = np.ones((batch_size, hidden_size), dtype=np.float32)
129 my_correct = np.zeros((batch_size, hidden_size), dtype=np.float32)
132 our_correct = np.zeros((batch_size, hidden_size), dtype=np.float32)
|
| /external/tensorflow/tensorflow/python/debug/examples/ |
| debug_mnist.py | 37 HIDDEN_SIZE = 500
92 hidden = nn_layer(x, IMAGE_SIZE**2, HIDDEN_SIZE, "hidden")
93 logits = nn_layer(hidden, HIDDEN_SIZE, NUM_LABELS, "output", tf.identity)
|
| /external/tensorflow/tensorflow/examples/learn/ |
| text_classification_character_rnn.py | 38 HIDDEN_SIZE = 20
48 cell = tf.nn.rnn_cell.GRUCell(HIDDEN_SIZE)
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.cc | 1140 int hidden_size() const { return hidden_size_; } function in class:perftools::gputools::cuda::CudnnRnnDescriptor
1417 int hidden_size = 0; member in struct:perftools::gputools::cuda::__anon40197::RnnModelDims
[all...] |
| cuda_dnn.h | 51 int num_layers, int hidden_size, int input_size,
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| stream_executor_pimpl.cc | 345 int num_layers, int hidden_size, int input_size,
355 num_layers, hidden_size, input_size, input_mode, direction_mode, rnn_mode,
|
| stream_executor_pimpl.h | 373 int num_layers, int hidden_size, int input_size,
[all...] |
| dnn.h | [all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| rnn.py | 469 rnn_cell = tf.nn.rnn_cell.BasicRNNCell(hidden_size)
[all...] |
