| /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, \
|