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 __anon652ace880702() local 323 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anon652ace880802() local
|