HomeSort by relevance Sort by last modified time
    Searched refs:cs_grad (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/contrib/rnn/python/ops/
lstm_ops.py 243 (_, cs_grad, _, _, _, _, h_grad) = grad
271 cs_grad,
304 cs_grad = grad[1]
325 cs_grad,
  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops_gpu.cu.cc 294 const T* cs_grad, // [batch_size, cell_size]
327 (one - co_local * co_local) * h_grad_local * o_local + cs_grad[cid];
366 typename TTypes<T>::ConstMatrix cs_grad,
384 co.data(), cs_grad.data(), h_grad.data(), do_.data(), dcs.data(),
440 typename TTypes<T>::ConstMatrix cs_grad, \
450 cs_grad, h_grad, do_, dcs, dci, df, di, dicfo, cs_prev_grad, wci_grad, \
lstm_ops.h 191 typename TTypes<T>::ConstMatrix cs_grad,
217 typename TTypes<T>::ConstMatrix cs_grad,
231 dcs.device(d) = (co.constant(T(1)) - co * co) * h_grad * o + cs_grad;
lstm_ops.cc 136 typename TTypes<T>::ConstMatrix cs_grad,
147 dcs.device(d) = (co.constant(T(1)) - co * co) * h_grad * o + cs_grad;
212 typename TTypes<T>::ConstMatrix cs_grad, \
222 i, cs, f, o, ci, co, cs_grad, h_grad, do_, dcs, dci, df, di, dicfo, \
468 OP_REQUIRES_OK(ctx, ctx->input("cs_grad", &cs_grad_tensor));
586 {"cs_grad"}, "cs_prev_grad",
683 typename TTypes<T>::ConstMatrix cs_grad, \
1118 const Tensor* cs_grad = nullptr; variable
    [all...]

Completed in 108 milliseconds