/external/tensorflow/tensorflow/contrib/rnn/ops/ |
D | lstm_ops.cc | 46 ShapeHandle x, cs_prev; in __anonad8b69780102() local 48 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anonad8b69780102() 51 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anonad8b69780102() 133 ShapeHandle x, cs_prev; in __anonad8b69780202() local 135 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anonad8b69780202() 138 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anonad8b69780202() 293 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anonad8b69780402() local 295 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &cs_prev)); in __anonad8b69780402() 304 c->set_output(1, cs_prev); in __anonad8b69780402()
|
/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 …]
|