Searched refs:LSTMBlockCell (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops.h | 104 struct LSTMBlockCell { struct
105 LSTMBlockCell(const int batch_size, const int input_size, const int cell_size) in LSTMBlockCell() argument
163 struct LSTMBlockCellFprop : public LSTMBlockCell { argument
166 : LSTMBlockCell(batch_size, input_size, cell_size) {} in LSTMBlockCellFprop()
188 struct LSTMBlockCellBprop : public LSTMBlockCell {
191 : LSTMBlockCell(batch_size, input_size, cell_size) {} in LSTMBlockCellBprop()
213 struct BlockLSTMBprop : public LSTMBlockCell {
216 : LSTMBlockCell(batch_size, input_size, cell_size) {} in BlockLSTMBprop()
|
| D | lstm_ops.cc | 46 const LSTMBlockCell& cell, OpKernelContext* ctx, const CPUDevice& d, in LSTMBlockCellFpropWithEigen()
131 const LSTMBlockCell& cell, OpKernelContext* ctx, const Device& d, in LSTMBlockCellBpropWithEigen()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BlockLSTMV2.pbtxt | 115 This is equivalent to applying LSTMBlockCell in a loop, like so:
132 Note that unlike LSTMBlockCell (and BlockLSTM) which uses ICFO gate layout,
|
| D | api_def_LSTMBlockCell.pbtxt | 2 graph_op_name: "LSTMBlockCell"
|
| D | api_def_LSTMBlockCellGrad.pbtxt | 138 This implementation is to be used in conjunction of LSTMBlockCell.
|
| D | api_def_BlockLSTM.pbtxt | 121 This is equivalent to applying LSTMBlockCell in a loop, like so:
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | LSTMBlockCell.pbtxt | 2 name: "LSTMBlockCell"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | LSTMBlockCell.pbtxt | 2 name: "LSTMBlockCell"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2080 name: "LSTMBlockCell"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2080 name: "LSTMBlockCell"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 43922 func LSTMBlockCell(scope *Scope, x tf.Output, cs_prev tf.Output, h_prev tf.Output, w tf.Output, wci… func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 20711 name: "LSTMBlockCell"
|