Searched defs:h_grad (Results 1 – 2 of 2) sorted by relevance
302 const T* h_grad, // [batch_size, cell_size] in lstm_gates_bprop()374 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_, in LSTMBlockCellBpropWithCUDA()
141 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_, in LSTMBlockCellBpropWithEigen()1141 const Tensor* h_grad = nullptr; in Compute() local