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

  /external/tensorflow/tensorflow/contrib/rnn/python/ops/
lstm_ops.py 51 wco=None,
69 wci = wcf = wco = 0
78 o = sigmoid(cs * wco + o)
96 wco: A `Tensor`. Must have the same type as `x`.
124 wco = wci
134 wco=wco,
151 wco=None,
167 wco: A `Tensor`. Must have the same type as `x`.
210 wco = wc
    [all...]
  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops_gpu.cu.cc 87 const T* wci, const T* wcf, const T* wco, T* o, T* h,
183 cs_local * wco[act_id]);
228 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b,
268 wco.data(), o.data(), h.data(), ci.data(), cs.data(), co.data(),
274 wco.data(), o.data(), h.data(), ci.data(), cs.data(), co.data(),
286 const T* wco, // [cell_size]
329 dcs_local += do_local * wco[act_id];
362 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b,
383 wco.data(), b.data(), i.data(), cs.data(), f.data(), o.data(), ci.data(),
417 typename TTypes<T>::ConstVec wcf, typename TTypes<T>::ConstVec wco, \
    [all...]
lstm_ops.h 164 typename TTypes<T>::ConstVec wco,
187 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b,
212 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b,
237 dcs + do_ * wco.reshape(p_shape).broadcast(p_broadcast_shape);
lstm_ops.cc 51 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b,
112 auto o_peep = cs * wco.reshape(p_shape).broadcast(p_broadcast_shape);
132 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b,
153 dcs + do_ * wco.reshape(p_shape).broadcast(p_broadcast_shape);
190 typename TTypes<T>::ConstVec wcf, typename TTypes<T>::ConstVec wco, \
198 h_prev, w, wci, wcf, wco, b, xh, i, cs, f, o, ci, co, icfo, h); \
207 typename TTypes<T>::ConstVec wcf, typename TTypes<T>::ConstVec wco, \
221 *this, ctx, d, use_peephole, x, cs_prev, h_prev, w, wci, wcf, wco, b, \
263 OP_REQUIRES_OK(ctx, ctx->input("wco", &wco_tensor));
392 typename TTypes<T>::ConstVec wcf, typename TTypes<T>::ConstVec wco, \
    [all...]
  /external/tensorflow/tensorflow/contrib/rnn/ops/
lstm_ops.cc 32 .Input("wco: T")
72 wci = wcf = wco = 0
81 o = sigmoid(cs * wco + o)
96 wco: The weight matrix for output gate peephole connection.
115 .Input("wco: T")
162 wco: The weight matrix for output gate peephole connection.
176 wco_grad: The gradient for wco to be back-propped.
187 .Input("wco: T")
226 x1, cs_prev, h_prev, w, wci, wcf, wco, b)
251 wco: The weight matrix for output gate peephole connection
    [all...]
  /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
lstm_ops_test.py 130 wco = variable_scope.get_variable(
161 wco=wco,
185 xs += [wci, wcf, wco]
194 xs += [wci, wcf, wco]

Completed in 613 milliseconds