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_gpu.cu.cc 245 const T* h_grad, // [batch_size, cell_size]
266 const T h_grad_local = h_grad[cid];
317 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_,
333 co.data(), cs_grad.data(), h_grad.data(), do_.data(), dcs.data(),
389 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_, \
398 cs_grad, h_grad, do_, dcs, dci, df, di, dicfo, cs_prev_grad, wci_grad, \
lstm_ops.h 190 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_,
216 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_,
226 do_.device(d) = o * (o.constant(T(1)) - o) * h_grad * co;
229 dcs.device(d) = (co.constant(T(1)) - co * co) * h_grad * o + cs_grad;
lstm_ops.cc 136 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_,
143 do_.device(d) = o * (o.constant(T(1)) - o) * h_grad * co;
146 dcs.device(d) = (co.constant(T(1)) - co * co) * h_grad * o + cs_grad;
211 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_, \
220 i, cs, f, o, ci, co, cs_grad, h_grad, do_, dcs, dci, df, di, dicfo, \
466 OP_REQUIRES_OK(ctx, ctx->input("h_grad", &h_grad_tensor));
679 typename TTypes<T>::ConstMatrix h_grad, typename TTypes<T>::Matrix do_, \
1112 const Tensor* h_grad = nullptr; variable
    [all...]
  /external/tensorflow/tensorflow/contrib/rnn/python/ops/
lstm_ops.py 239 (_, cs_grad, _, _, _, _, h_grad) = grad
268 h_grad,
301 h_grad = grad[6]
322 h_grad,

Completed in 120 milliseconds