/external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
D | lstm_ops.py | 46 h_prev, argument 130 h_prev=h_prev, 148 h_prev=None, argument 200 if cs_prev is None or h_prev is None: 205 if h_prev is None: 206 h_prev = zero_state 217 h_prev=h_prev, 241 (x, cs_prev, h_prev, w, wci, wcf, wco, b) = op.inputs 259 h_prev, 283 h_prev_grad.get_shape().merge_with(h_prev.get_shape()) [all …]
|
D | gru_ops.py | 83 x, h_prev, w_ru, w_c, b_ru, b_c = op.inputs 88 x, h_prev, w_ru, w_c, b_ru, b_c, r, u, c, d_h) 90 x_h_prev = array_ops.concat([x, h_prev], 1) 94 x_h_prevr = array_ops.concat([x, h_prev * r], 1) 197 def call(self, inputs, h_prev): argument 200 cell_size = h_prev.get_shape().with_rank(2)[1] 208 h_prev=h_prev,
|
D | rnn_cell.py | 2002 (c_prev, h_prev) = state 2005 in_mask_gates = [x, h_prev] 2019 self._linear2 = _Linear([x, h_prev], self._num_units, True) 2020 new_input = math_ops.tanh(self._linear2([x, h_prev])) 2024 in_out_gate = [x, h_prev] 2058 new_h = k * new_h + (1 - k) * h_prev
|
/external/tensorflow/tensorflow/contrib/rnn/ops/ |
D | gru_ops.cc | 36 ShapeHandle x, h_prev; in __anon6fce7f260102() local 38 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &h_prev)); in __anon6fce7f260102() 41 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon6fce7f260102() 114 ShapeHandle x, h_prev, w_ru; in __anon6fce7f260202() local 116 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &h_prev)); in __anon6fce7f260202() 120 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon6fce7f260202()
|
D | lstm_ops.cc | 293 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anonad8b69780402() local 296 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 2, &h_prev)); in __anonad8b69780402() 305 c->set_output(2, h_prev); in __anonad8b69780402()
|
/external/tensorflow/tensorflow/contrib/rnn/kernels/ |
D | gru_ops.h | 76 typename TTypes<T>::ConstMatrix h_prev, in operator() 85 x_h_prev.slice(h_offsets(), h_extends()).device(d) = h_prev; in operator() 106 x_h_prevr.slice(h_offsets(), h_extends()).device(d) = h_prev * r; in operator() 120 h.device(d) = u * (h_prev - c) + c; in operator() 132 typename TTypes<T>::ConstMatrix h_prev, in operator() 149 d_u_bar.device(d) = d_h * (h_prev - c) * u * (u.constant(T(1)) - u); in operator() 160 d_r_bar.device(d) = (d_hr * h_prev * r) * (r.constant(T(1)) - r); in operator()
|
D | lstm_ops_gpu.cu.cc | 194 __global__ void concat_xh(T* xh, const T* x, const T* h_prev, in concat_xh() argument 217 xh[gid] = h_prev[output_row * cell_size + output_col - input_size]; in concat_xh() 226 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellFpropWithCUDA() argument 246 xh.data(), x.data(), h_prev.data(), batch_size, in LSTMBlockCellFpropWithCUDA() 282 const T* h_prev, // [batch_size, cell_size] in lstm_gates_bprop() argument 360 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellBpropWithCUDA() argument 382 cs_prev.data(), h_prev.data(), w.data(), wci.data(), wcf.data(), in LSTMBlockCellBpropWithCUDA() 415 typename TTypes<T>::ConstMatrix h_prev, \ 424 use_peephole, x, cs_prev, h_prev, w, wci, \ 433 typename TTypes<T>::ConstMatrix h_prev, \ [all …]
|
D | lstm_ops.h | 160 typename TTypes<T>::ConstMatrix h_prev, 185 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, 210 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in operator() 270 xh.slice(xh_h_offsets(), xh_h_extents()).device(d) = h_prev; in operator()
|
D | lstm_ops.cc | 49 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellFpropWithEigen() argument 59 xh.slice(cell.xh_h_offsets(), cell.xh_h_extents()).device(d) = h_prev; in LSTMBlockCellFpropWithEigen() 130 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellBpropWithEigen() argument 188 typename TTypes<T>::ConstMatrix h_prev, \ 198 h_prev, w, wci, wcf, wco, b, xh, i, cs, f, o, ci, co, icfo, h); \ 205 typename TTypes<T>::ConstMatrix h_prev, \ 221 *this, ctx, d, use_peephole, x, cs_prev, h_prev, w, wci, wcf, wco, b, \ 390 typename TTypes<T>::ConstMatrix h_prev, \ 676 typename TTypes<T>::ConstMatrix h_prev, \ 1318 typename TTypes<T>::ConstMatrix h_prev, \
|
D | gru_ops.cc | 393 typename TTypes<T>::ConstMatrix h_prev, \
|