Searched defs:h_prev (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | rnn_ops.cc | 38 ShapeHandle x, h_prev; in __anon4a6db9170102() local
68 ShapeHandle x, h_prev, w_ru; in __anon4a6db9170202() local
271 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anon4a6db9170702() local
323 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anon4a6db9170802() local
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops_gpu.cu.cc | 199 __global__ void concat_xh(T* xh, const T* x, const T* h_prev, in concat_xh()
231 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellFpropWithCUDA()
289 const T* h_prev, // [batch_size, cell_size] in lstm_gates_bprop()
367 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellBpropWithCUDA()
|
| D | lstm_ops.cc | 49 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellFpropWithEigen()
134 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellBpropWithEigen()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_grad_test.py | 119 def _lstm_block(self, op, w, b, x, cs_prev, h_prev): argument
|