Home
last modified time | relevance | path

Searched defs:cs_grad (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dlstm_ops_gpu.cu.cc294 const T* cs_grad, // [batch_size, cell_size] in lstm_gates_bprop()
366 typename TTypes<T>::ConstMatrix cs_grad, in LSTMBlockCellBpropWithCUDA()
Dlstm_ops.cc136 typename TTypes<T>::ConstMatrix cs_grad, in LSTMBlockCellBpropWithEigen()
1118 const Tensor* cs_grad = nullptr; in Compute() local