Searched defs:cs_grad (Results 1 – 2 of 2) sorted by relevance
301 const T* cs_grad, // [batch_size, cell_size] in lstm_gates_bprop()373 typename TTypes<T>::ConstMatrix cs_grad, in LSTMBlockCellBpropWithCUDA()
140 typename TTypes<T>::ConstMatrix cs_grad, in LSTMBlockCellBpropWithEigen()1220 const Tensor* cs_grad = nullptr; in Compute() local