Searched defs:wco (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()
233 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellFpropWithCUDA()
293 const T* wco, // [cell_size] in lstm_gates_bprop()
369 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellBpropWithCUDA()
|
| D | lstm_ops.cc | 51 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellFpropWithEigen()
136 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellBpropWithEigen()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | rnn_ops.cc | 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
|