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

  /external/tensorflow/tensorflow/contrib/rnn/ops/
lstm_ops.cc 27 .Input("cs_prev: T")
46 ShapeHandle x, cs_prev;
48 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev));
51 DimensionHandle cell_size = c->Dim(cs_prev, 1);
74 i = sigmoid(cs_prev * wci + i)
75 f = sigmoid(cs_prev * wcf + f)
78 cs = ci .* i + cs_prev .* f
91 cs_prev: Value of the cell state at previous time step.
110 .Input("cs_prev: T")
133 ShapeHandle x, cs_prev;
    [all...]
  /external/tensorflow/tensorflow/contrib/rnn/python/ops/
lstm_ops.py 45 cs_prev,
71 i = sigmoid(cs_prev * wci + i)
72 f = sigmoid(cs_prev * wcf + f)
75 cs = ci .* i + cs_prev .* f
86 cs_prev: A `Tensor`. Must have the same type as `x`.
119 cell_size = cs_prev.get_shape().with_rank(2).dims[1].value
121 raise ValueError("cell_size from `cs_prev` should not be None.")
129 cs_prev=cs_prev,
147 cs_prev=None
    [all...]
  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops_gpu.cu.cc 86 __global__ void lstm_gates(const T* icfo, const T* b, const T* cs_prev,
151 cs_prev[cid] * wci[act_id]);
164 forget_bias_t + cs_prev[cid] * wcf[act_id]);
171 T cs_local = i_local * ci_local + f_local * cs_prev[cid];
225 typename TTypes<T>::ConstMatrix cs_prev,
267 icfo.data(), b.data(), cs_prev.data(), wci.data(), wcf.data(),
273 icfo.data(), b.data(), cs_prev.data(), wci.data(), wcf.data(),
281 const T* cs_prev, // [batch_size, cell_size]
338 const T df_local = f_local * (one - f_local) * dcs_local * cs_prev[cid];
359 typename TTypes<T>::ConstMatrix cs_prev,
    [all...]
lstm_ops.h 159 typename TTypes<T>::ConstMatrix cs_prev,
184 typename TTypes<T>::ConstMatrix cs_prev,
209 typename TTypes<T>::ConstMatrix cs_prev,
244 df.device(d) = f * (f.constant(T(1)) - f) * dcs * cs_prev;
285 wci_grad.device(d) += (di * cs_prev).sum(Eigen::array<int, 1>({0}));
286 wcf_grad.device(d) += (df * cs_prev).sum(Eigen::array<int, 1>({0}));
lstm_ops.cc 48 typename TTypes<T>::ConstMatrix x, typename TTypes<T>::ConstMatrix cs_prev,
75 auto i_peep = cs_prev * wci.reshape(p_shape).broadcast(p_broadcast_shape);
89 auto f_peep = cs_prev * wcf.reshape(p_shape).broadcast(p_broadcast_shape);
99 // cs = ci .* i + f .* cs_prev
100 cs.device(d) = i * ci + f * cs_prev;
129 typename TTypes<T>::ConstMatrix cs_prev,
160 df.device(d) = f * (f.constant(T(1)) - f) * dcs * cs_prev;
175 wci_grad.device(d) = (di * cs_prev).sum(Eigen::array<int, 1>({0}));
176 wcf_grad.device(d) = (df * cs_prev).sum(Eigen::array<int, 1>({0}));
187 typename TTypes<T>::ConstMatrix cs_prev, \
    [all...]

Completed in 182 milliseconds