| /external/tensorflow/tensorflow/contrib/rnn/ops/ |
| D | lstm_ops.cc | 46 ShapeHandle x, cs_prev; in __anon4050c6010102() local
48 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anon4050c6010102()
51 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon4050c6010102()
133 ShapeHandle x, cs_prev; in __anon4050c6010202() local
135 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anon4050c6010202()
138 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon4050c6010202()
293 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anon4050c6010402() local
295 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &cs_prev)); in __anon4050c6010402()
304 c->set_output(1, cs_prev); in __anon4050c6010402()
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | lstm_ops.py | 45 cs_prev, argument
119 cell_size = cs_prev.get_shape().with_rank(2).dims[1].value
129 cs_prev=cs_prev,
147 cs_prev=None, argument
200 if cs_prev is None or h_prev is None:
203 if cs_prev is None:
204 cs_prev = zero_state
216 cs_prev=cs_prev,
241 (x, cs_prev, h_prev, w, wci, wcf, wco, b) = op.inputs
251 cell_size = cs_prev.get_shape().with_rank(2).dims[1].value
[all …]
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | lstm_ops_gpu.cu.cc | 86 __global__ void lstm_gates(const T* icfo, const T* b, const T* cs_prev, in lstm_gates() argument
151 cs_prev[cid] * wci[act_id]); in lstm_gates()
164 forget_bias_t + cs_prev[cid] * wcf[act_id]); in lstm_gates()
171 T cs_local = i_local * ci_local + f_local * cs_prev[cid]; in lstm_gates()
225 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellFpropWithCUDA() argument
267 icfo.data(), b.data(), cs_prev.data(), wci.data(), wcf.data(), in LSTMBlockCellFpropWithCUDA()
273 icfo.data(), b.data(), cs_prev.data(), wci.data(), wcf.data(), in LSTMBlockCellFpropWithCUDA()
281 const T* cs_prev, // [batch_size, cell_size] in lstm_gates_bprop() argument
338 const T df_local = f_local * (one - f_local) * dcs_local * cs_prev[cid]; in lstm_gates_bprop()
359 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellBpropWithCUDA() argument
[all …]
|
| D | lstm_ops.h | 159 typename TTypes<T>::ConstMatrix cs_prev,
184 typename TTypes<T>::ConstMatrix cs_prev,
209 typename TTypes<T>::ConstMatrix cs_prev, in operator()
244 df.device(d) = f * (f.constant(T(1)) - f) * dcs * cs_prev; in operator()
285 wci_grad.device(d) += (di * cs_prev).sum(Eigen::array<int, 1>({0})); in operator()
286 wcf_grad.device(d) += (df * cs_prev).sum(Eigen::array<int, 1>({0})); in operator()
|
| D | lstm_ops.cc | 48 typename TTypes<T>::ConstMatrix x, typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellFpropWithEigen() argument
75 auto i_peep = cs_prev * wci.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellFpropWithEigen()
89 auto f_peep = cs_prev * wcf.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellFpropWithEigen()
100 cs.device(d) = i * ci + f * cs_prev; in LSTMBlockCellFpropWithEigen()
129 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellBpropWithEigen() argument
160 df.device(d) = f * (f.constant(T(1)) - f) * dcs * cs_prev; in LSTMBlockCellBpropWithEigen()
175 wci_grad.device(d) = (di * cs_prev).sum(Eigen::array<int, 1>({0})); in LSTMBlockCellBpropWithEigen()
176 wcf_grad.device(d) = (df * cs_prev).sum(Eigen::array<int, 1>({0})); in LSTMBlockCellBpropWithEigen()
187 typename TTypes<T>::ConstMatrix cs_prev, \
197 *this, ctx, d, forget_bias, cell_clip, use_peephole, x, cs_prev, \
[all …]
|