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

  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops.h 192 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_,
218 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_,
228 do_.device(d) = o * (o.constant(T(1)) - o) * h_grad * co;
231 dcs.device(d) = (co.constant(T(1)) - co * co) * h_grad * o + cs_grad;
lstm_ops_gpu.cu.cc 295 const T* h_grad, // [batch_size, cell_size]
316 const T h_grad_local = h_grad[cid];
367 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_,
384 co.data(), cs_grad.data(), h_grad.data(), do_.data(), dcs.data(),
441 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_, \
450 cs_grad, h_grad, do_, dcs, dci, df, di, dicfo, cs_prev_grad, wci_grad, \
lstm_ops.cc 137 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_,
144 do_.device(d) = o * (o.constant(T(1)) - o) * h_grad * co;
147 dcs.device(d) = (co.constant(T(1)) - co * co) * h_grad * o + cs_grad;
213 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_, \
222 i, cs, f, o, ci, co, cs_grad, h_grad, do_, dcs, dci, df, di, dicfo, \
471 OP_REQUIRES_OK(ctx, ctx->input("h_grad", &h_grad_tensor));
684 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_, \
1121 const Tensor* h_grad = nullptr; variable
    [all...]
  /external/tensorflow/tensorflow/contrib/rnn/python/ops/
lstm_ops.py 243 (_, cs_grad, _, _, _, _, h_grad) = grad
272 h_grad,
305 h_grad = grad[6]
326 h_grad,

Completed in 468 milliseconds