Home
last modified time | relevance | path

Searched refs:LSTMBlockCell (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dlstm_ops.h94 struct LSTMBlockCell { struct
95 LSTMBlockCell(const int batch_size, const int input_size, const int cell_size) in LSTMBlockCell() function
151 struct LSTMBlockCellFprop : public LSTMBlockCell { argument
154 : LSTMBlockCell(batch_size, input_size, cell_size) {} in LSTMBlockCellFprop()
176 struct LSTMBlockCellBprop : public LSTMBlockCell {
179 : LSTMBlockCell(batch_size, input_size, cell_size) {} in LSTMBlockCellBprop()
201 struct BlockLSTMBprop : public LSTMBlockCell {
204 : LSTMBlockCell(batch_size, input_size, cell_size) {} in BlockLSTMBprop()
Dlstm_ops.cc46 const LSTMBlockCell& cell, OpKernelContext* ctx, const CPUDevice& d, in LSTMBlockCellFpropWithEigen()
127 const LSTMBlockCell& cell, OpKernelContext* ctx, const Device& d, in LSTMBlockCellBpropWithEigen()
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
Dlstm_ops_test.py225 cell = lstm_ops.LSTMBlockCell(cell_size)
245 [lstm_ops.LSTMBlockCell(2)
276 cell = lstm_ops.LSTMBlockCell(10)
277 pcell = lstm_ops.LSTMBlockCell(10, use_peephole=True)
335 [lstm_ops.LSTMBlockCell(2)
388 [lstm_ops.LSTMBlockCell(2, use_peephole=True) for _ in range(2)],
548 cell = lstm_ops.LSTMBlockCell(config["cell_size"], dtype=dtype)
610 cell = lstm_ops.LSTMBlockCell(cell_size, dtype=dtype)
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_ops_benchmark.py157 [lstm_ops.LSTMBlockCell(num_units) for _ in range(num_layers)])
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Dlstm_ops.py335 class LSTMBlockCell(LayerRNNCell): class
375 super(LSTMBlockCell, self).__init__(_reuse=reuse, dtype=dtype, name=name)
/external/tensorflow/tensorflow/contrib/autograph/examples/notebooks/
Drnn_keras_estimator.ipynb189 " self.lower_cell = tf.contrib.rnn.LSTMBlockCell(256, dtype=tf.float32)\n",
190 " self.upper_cell = tf.contrib.rnn.LSTMBlockCell(128, dtype=tf.float32)\n",
198 " cell: An object of type tf.contrib.rnn.LSTMBlockCell\n",
Ddev_summit_2018_demo.ipynb1106 " lower_cell = tf.contrib.rnn.LSTMBlockCell(256)\n",
1108 " upper_cell = tf.contrib.rnn.LSTMBlockCell(128)\n",
1120 " cell: An object of type tf.contrib.rnn.LSTMBlockCell\n",
1152 " lower_cell: An object of type tf.contrib.rnn.LSTMBlockCell\n",
1153 " upper_cell: An object of type tf.contrib.rnn.LSTMBlockCell\n",
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py61 class CudnnCompatibleLSTMCell(lstm_ops.LSTMBlockCell):