Searched defs:h_prev (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/contrib/rnn/ops/ |
| D | gru_ops.cc | 36 ShapeHandle x, h_prev; in __anon07a5ee8f0102() local
114 ShapeHandle x, h_prev, w_ru; in __anon07a5ee8f0202() local
|
| D | lstm_ops.cc | 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 | 194 __global__ void concat_xh(T* xh, const T* x, const T* h_prev, in concat_xh()
226 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellFpropWithCUDA()
282 const T* h_prev, // [batch_size, cell_size] in lstm_gates_bprop()
360 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()
130 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellBpropWithEigen()
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | gru_ops.py | 197 def call(self, inputs, h_prev): argument
|
| D | lstm_ops.py | 46 h_prev, argument
148 h_prev=None, argument
|