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