OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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