Searched defs:wci (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops_gpu.cu.cc | 86 const T* wci, const T* wcf, const T* wco, T* o, T* h, in lstm_gates()
232 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellFpropWithCUDA()
291 const T* wci, // [cell_size] in lstm_gates_bprop()
368 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellBpropWithCUDA()
|
| D | lstm_ops.cc | 50 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellFpropWithEigen()
135 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellBpropWithEigen()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | rnn_ops.cc | 271 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anona90d3fa60702() local
323 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anona90d3fa60802() local
|