| /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])
79 w_peephole = array_ops.zeros(cs_prev.shape[1:], dtype=w.dtype)
89 cs_prev=cs_prev,
102 cs_prev=cs_prev,
123 def _lstm_block(self, op, w, b, x, cs_prev, h_prev): argument
124 w_peephole = array_ops.zeros(cs_prev.shape[1:], dtype=w.dtype)
128 cs_prev=cs_prev,
|
| D | rnn_grad.py | 26 seq_len_max, x, cs_prev, h_prev, w, wci, wcf, wco, b = op.inputs
33 cs_prev=cs_prev,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | rnn_ops.cc | 106 ShapeHandle x, cs_prev; in __anon652ace880302() local
108 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anon652ace880302()
111 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon652ace880302()
144 ShapeHandle x, cs_prev; in __anon652ace880402() local
146 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anon652ace880402()
149 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon652ace880402()
271 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anon652ace880702() local
273 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &cs_prev)); in __anon652ace880702()
282 c->set_output(1, cs_prev); in __anon652ace880702()
323 ShapeHandle x, cs_prev, h_prev, w, wci, wco, wcf, b; in __anon652ace880802() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops_gpu.cu.cc | 85 __global__ void lstm_gates(const T* gates, const T* b, const T* cs_prev, in lstm_gates() argument
152 cs_prev[cid] * wci[act_id]); in lstm_gates()
168 forget_bias_t + cs_prev[cid] * wcf[act_id]); in lstm_gates()
175 T cs_local = i_local * ci_local + f_local * cs_prev[cid]; in lstm_gates()
230 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellFpropWithCUDA() argument
272 cu_stream, gates.data(), b.data(), cs_prev.data(), wci.data(), in LSTMBlockCellFpropWithCUDA()
279 cu_stream, gates.data(), b.data(), cs_prev.data(), wci.data(), in LSTMBlockCellFpropWithCUDA()
288 const T* cs_prev, // [batch_size, cell_size] in lstm_gates_bprop() argument
345 const T df_local = f_local * (one - f_local) * dcs_local * cs_prev[cid]; in lstm_gates_bprop()
366 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellBpropWithCUDA() argument
[all …]
|
| D | lstm_ops.h | 171 typename TTypes<T>::ConstMatrix cs_prev,
196 typename TTypes<T>::ConstMatrix cs_prev,
221 typename TTypes<T>::ConstMatrix cs_prev, in operator()
257 df.device(d) = f * (f.constant(T(1)) - f) * dcs * cs_prev; in operator()
298 wci_grad.device(d) += (di * cs_prev).sum(Eigen::array<int, 1>({0})); in operator()
299 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()
91 auto f_peep = cs_prev * wcf.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellFpropWithEigen()
104 cs.device(d) = i * ci + f * cs_prev; in LSTMBlockCellFpropWithEigen()
133 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellBpropWithEigen() argument
164 df.device(d) = f * (f.constant(T(1)) - f) * dcs * cs_prev; in LSTMBlockCellBpropWithEigen()
181 wci_grad.device(d) = (di * cs_prev).sum(Eigen::array<int, 1>({0})); in LSTMBlockCellBpropWithEigen()
182 wcf_grad.device(d) = (df * cs_prev).sum(Eigen::array<int, 1>({0})); in LSTMBlockCellBpropWithEigen()
194 typename TTypes<T>::ConstMatrix cs_prev, \
204 *this, ctx, d, forget_bias, cell_clip, use_peephole, x, cs_prev, \
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_LSTMBlockCell.pbtxt | 11 name: "cs_prev"
127 i = sigmoid(cs_prev * wci + i)
128 f = sigmoid(cs_prev * wcf + f)
131 cs = ci .* i + cs_prev .* f
|
| D | api_def_BlockLSTM.pbtxt | 18 name: "cs_prev"
126 x1, cs_prev, h_prev, w, wci, wcf, wco, b)
127 cs_prev = cs1
|
| D | api_def_BlockLSTMV2.pbtxt | 18 name: "cs_prev"
120 x1, cs_prev, h_prev, w, wci, wcf, wco, b)
121 cs_prev = cs1
|
| D | api_def_BlockLSTMGradV2.pbtxt | 18 name: "cs_prev"
122 The gradient of cs_prev to be back-propped.
|
| D | api_def_BlockLSTMGrad.pbtxt | 18 name: "cs_prev"
122 The gradient of cs_prev to be back-propped.
|
| D | api_def_LSTMBlockCellGrad.pbtxt | 11 name: "cs_prev"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | LSTMBlockCell.pbtxt | 8 name: "cs_prev"
|
| D | BlockLSTMV2.pbtxt | 12 name: "cs_prev"
|
| D | BlockLSTM.pbtxt | 12 name: "cs_prev"
|
| D | LSTMBlockCellGrad.pbtxt | 8 name: "cs_prev"
|
| D | BlockLSTMGrad.pbtxt | 12 name: "cs_prev"
|
| D | BlockLSTMGradV2.pbtxt | 12 name: "cs_prev"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | LSTMBlockCell.pbtxt | 8 name: "cs_prev"
|
| D | BlockLSTMV2.pbtxt | 12 name: "cs_prev"
|
| D | BlockLSTM.pbtxt | 12 name: "cs_prev"
|
| D | LSTMBlockCellGrad.pbtxt | 8 name: "cs_prev"
|
| D | BlockLSTMGrad.pbtxt | 12 name: "cs_prev"
|
| D | BlockLSTMGradV2.pbtxt | 12 name: "cs_prev"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 521 …argspec: "args=[\'seq_len_max\', \'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\',…
525 …argspec: "args=[\'seq_len_max\', \'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\',…
529 …argspec: "args=[\'seq_len_max\', \'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\',…
533 …argspec: "args=[\'seq_len_max\', \'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\',…
2081 …argspec: "args=[\'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\', \'b\', \'forget_…
2085 …argspec: "args=[\'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\', \'b\', \'i\', \'…
|