Home
last modified time | relevance | path

Searched defs:wci (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/rnn/
Dlstm_ops_gpu.cu.cc86 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()
Dlstm_ops.cc50 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/
Drnn_ops.cc271 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