| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_grad_test.py | 48 cs_prev = h_prev = deterministic_random_uniform([batch_size, hidden_size])
55 w, b, x, cs_prev, h_prev)
60 gen_rnn_ops.BlockLSTMV2, w_ifco, b_ifco, x, cs_prev, h_prev)
78 cs_prev = h_prev = deterministic_random_uniform([batch_size, hidden_size])
90 h_prev=h_prev,
103 h_prev=h_prev,
123 def _lstm_block(self, op, w, b, x, cs_prev, h_prev): argument
129 h_prev=h_prev,
|
| D | rnn_grad.py | 26 seq_len_max, x, cs_prev, h_prev, w, wci, wcf, wco, b = op.inputs
34 h_prev=h_prev,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | rnn_ops.cc | 38 ShapeHandle x, h_prev; in __anon652ace880102() local
40 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &h_prev)); in __anon652ace880102()
43 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon652ace880102()
68 ShapeHandle x, h_prev, w_ru; in __anon652ace880202() local
70 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &h_prev)); in __anon652ace880202()
74 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon652ace880202()
271 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anon652ace880702() local
274 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 2, &h_prev)); in __anon652ace880702()
283 c->set_output(2, h_prev); in __anon652ace880702()
323 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anon652ace880802() local
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_GRUBlockCell.pbtxt | 8 h_prev: State input from the previous GRU cell.
34 x_h_prev = [x, h_prev]
41 h_prevr = h_prev \circ r
48 h = (1-u) \circ c + u \circ h_prev
|
| D | api_def_BlockLSTM.pbtxt | 24 name: "h_prev"
126 x1, cs_prev, h_prev, w, wci, wcf, wco, b)
128 h_prev = h1
|
| D | api_def_BlockLSTMV2.pbtxt | 24 name: "h_prev"
120 x1, cs_prev, h_prev, w, wci, wcf, wco, b)
122 h_prev = h1
|
| D | api_def_LSTMBlockCell.pbtxt | 17 name: "h_prev"
120 xh = [x, h_prev]
|
| D | api_def_BlockLSTMGradV2.pbtxt | 24 name: "h_prev"
128 The gradient of h_prev to be back-propped.
|
| D | api_def_BlockLSTMGrad.pbtxt | 24 name: "h_prev"
128 The gradient of h_prev to be back-propped.
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| 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 | 199 __global__ void concat_xh(T* xh, const T* x, const T* h_prev, in concat_xh() argument
222 xh[gid] = h_prev[output_row * cell_size + output_col - input_size]; in concat_xh()
231 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellFpropWithCUDA() argument
251 xh.data(), x.data(), h_prev.data(), batch_size, in LSTMBlockCellFpropWithCUDA()
289 const T* h_prev, // [batch_size, cell_size] in lstm_gates_bprop() argument
367 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellBpropWithCUDA() argument
389 cs_prev.data(), h_prev.data(), w.data(), wci.data(), wcf.data(), in LSTMBlockCellBpropWithCUDA()
417 typename TTypes<T>::ConstMatrix h_prev, \
426 ctx, d, forget_bias, cell_clip, use_peephole, x, cs_prev, h_prev, w, \
436 typename TTypes<T>::ConstMatrix h_prev, \
[all …]
|
| D | lstm_ops.h | 172 typename TTypes<T>::ConstMatrix h_prev,
197 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w,
222 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in operator()
283 xh.slice(xh_h_offsets(), xh_h_extents()).device(d) = h_prev; in operator()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | GRUBlockCell.pbtxt | 8 name: "h_prev"
|
| D | GRUBlockCellGrad.pbtxt | 8 name: "h_prev"
|
| D | LSTMBlockCell.pbtxt | 12 name: "h_prev"
|
| D | BlockLSTMV2.pbtxt | 16 name: "h_prev"
|
| D | BlockLSTM.pbtxt | 16 name: "h_prev"
|
| D | LSTMBlockCellGrad.pbtxt | 12 name: "h_prev"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | GRUBlockCell.pbtxt | 8 name: "h_prev"
|
| D | GRUBlockCellGrad.pbtxt | 8 name: "h_prev"
|
| D | LSTMBlockCell.pbtxt | 12 name: "h_prev"
|
| D | BlockLSTMV2.pbtxt | 16 name: "h_prev"
|
| D | BlockLSTM.pbtxt | 16 name: "h_prev"
|
| D | LSTMBlockCellGrad.pbtxt | 12 name: "h_prev"
|
| D | BlockLSTMGrad.pbtxt | 16 name: "h_prev"
|