Searched defs:cs_prev (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | rnn_ops.cc | 106 ShapeHandle x, cs_prev; in __anon652ace880302() local
144 ShapeHandle x, cs_prev; in __anon652ace880402() local
271 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anon652ace880702() local
323 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anon652ace880802() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_grad_test.py | 123 def _lstm_block(self, op, w, b, x, cs_prev, h_prev): argument
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops_gpu.cu.cc | 85 __global__ void lstm_gates(const T* gates, const T* b, const T* cs_prev, in lstm_gates()
230 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellFpropWithCUDA()
288 const T* cs_prev, // [batch_size, cell_size] in lstm_gates_bprop()
366 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()
133 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellBpropWithEigen()
|