Searched defs:cs_prev (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/rnn/ops/ |
| D | lstm_ops.cc | 46 ShapeHandle x, cs_prev; in __anon4050c6010102() local
133 ShapeHandle x, cs_prev; in __anon4050c6010202() local
293 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anon4050c6010402() local
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | lstm_ops_gpu.cu.cc | 86 __global__ void lstm_gates(const T* icfo, const T* b, const T* cs_prev, in lstm_gates()
225 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellFpropWithCUDA()
281 const T* cs_prev, // [batch_size, cell_size] in lstm_gates_bprop()
359 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellBpropWithCUDA()
|
| D | lstm_ops.cc | 48 typename TTypes<T>::ConstMatrix x, typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellFpropWithEigen()
129 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellBpropWithEigen()
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | lstm_ops.py | 45 cs_prev, argument
147 cs_prev=None, argument
|