HomeSort by relevance Sort by last modified time
    Searched refs:LSTMBlockCell (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops.h 94 struct LSTMBlockCell {
95 LSTMBlockCell(const int batch_size, const int input_size, const int cell_size)
151 struct LSTMBlockCellFprop : public LSTMBlockCell {
154 : LSTMBlockCell(batch_size, input_size, cell_size) {}
176 struct LSTMBlockCellBprop : public LSTMBlockCell {
179 : LSTMBlockCell(batch_size, input_size, cell_size) {}
201 struct BlockLSTMBprop : public LSTMBlockCell {
204 : LSTMBlockCell(batch_size, input_size, cell_size) {}
lstm_ops.cc 46 const LSTMBlockCell& cell, OpKernelContext* ctx, const CPUDevice& d,
127 const LSTMBlockCell& cell, OpKernelContext* ctx, const Device& d,
375 Name("LSTMBlockCell").Device(DEVICE_CPU).TypeConstraint<T>("T"), \
408 Name("LSTMBlockCell").Device(DEVICE_GPU).TypeConstraint<T>("T"), \
    [all...]
  /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
lstm_ops_test.py 225 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)],
528 print("LSTMBlockCell Seconds per inference.")
548 cell = lstm_ops.LSTMBlockCell(config["cell_size"], dtype=dtype)
579 print("LSTMBlockCell Seconds per inference.")
610 cell = lstm_ops.LSTMBlockCell(cell_size, dtype=dtype
    [all...]
  /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
cudnn_rnn_ops_benchmark.py 157 [lstm_ops.LSTMBlockCell(num_units) for _ in range(num_layers)])
  /external/tensorflow/tensorflow/contrib/rnn/python/ops/
lstm_ops.py 238 @ops.RegisterGradient("LSTMBlockCell")
240 """Gradient for LSTMBlockCell."""
335 class LSTMBlockCell(LayerRNNCell):
375 super(LSTMBlockCell, self).__init__(_reuse=reuse, dtype=dtype, name=name)
590 LSTMBlockCell defined above.
  /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
cudnn_rnn_ops.py 61 class CudnnCompatibleLSTMCell(lstm_ops.LSTMBlockCell):
64 A simple wrapper around `tf.contrib.rnn.LSTMBlockCell` to use along with
    [all...]

Completed in 122 milliseconds