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

  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops_gpu.cu.cc 302 T* cs_prev_grad, // [batch_size, cell_size]
350 cs_prev_grad[cid] = dcs_local * f_local;
352 cs_prev_grad[cid] += di_local * wci[act_id] + df_local * wcf[act_id];
370 typename TTypes<T>::Matrix dicfo, typename TTypes<T>::Matrix cs_prev_grad,
385 dci.data(), df.data(), di.data(), dicfo.data(), cs_prev_grad.data(),
391 cs_prev_grad.device(d) =
392 cs_prev_grad + di * wci.reshape(p_shape).broadcast(p_broadcast_shape) +
445 typename TTypes<T>::Matrix cs_prev_grad, \
450 cs_grad, h_grad, do_, dcs, dci, df, di, dicfo, cs_prev_grad, wci_grad, \
lstm_ops.h 195 typename TTypes<T>::Matrix dicfo, typename TTypes<T>::Matrix cs_prev_grad,
221 typename TTypes<T>::Matrix dicfo, typename TTypes<T>::Matrix cs_prev_grad,
254 cs_prev_grad.device(d) = dcs * f;
256 cs_prev_grad.device(d) =
257 cs_prev_grad +
lstm_ops.cc 140 typename TTypes<T>::Matrix dicfo, typename TTypes<T>::Matrix cs_prev_grad,
170 cs_prev_grad.device(d) = dcs * f;
172 cs_prev_grad.device(d) =
173 cs_prev_grad + di * wci.reshape(p_shape).broadcast(p_broadcast_shape) +
217 typename TTypes<T>::Matrix cs_prev_grad, \
223 cs_prev_grad, wci_grad, wcf_grad, wco_grad); \
586 {"cs_grad"}, "cs_prev_grad",
688 typename TTypes<T>::Matrix cs_prev_grad, \
    [all...]
  /external/tensorflow/tensorflow/contrib/rnn/python/ops/
lstm_ops.py 235 _lstm_block_cell_grad_outputs = ["cs_prev_grad", "dicfo"]
255 (cs_prev_grad, dicfo, wci_grad, wcf_grad,
294 return (x_grad, cs_prev_grad, h_prev_grad, w_grad, wci_grad, wcf_grad,
307 (x_grad, cs_prev_grad, h_prev_grad, w_grad, wci_grad, wcf_grad, wco_grad,
330 None, x_grad, cs_prev_grad, h_prev_grad, w_grad, wci_grad, wcf_grad,

Completed in 211 milliseconds