| /external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/ |
| grid_rnn_cell.py | 378 forget_bias=1,
382 return rnn.BasicLSTMCell(num_units=n, forget_bias=forget_bias)
409 forget_bias=1,
413 return rnn.BasicLSTMCell(num_units=n, forget_bias=forget_bias)
438 forget_bias=1.0,
444 num_units=n, forget_bias=forget_bias, use_peepholes=use_peepholes)
471 forget_bias=1.0
[all...] |
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| lstm_ops.py | 52 forget_bias=None,
66 f = f + forget_bias
98 forget_bias: An optional `float`. Defaults to `1`. The forget gate bias.
136 forget_bias=forget_bias,
152 forget_bias=None,
168 forget_bias: An optional `float`. Defaults to `1`.
223 forget_bias=forget_bias,
340 We add `forget_bias` (default: 1) to the biases of the forget gate in order t
[all...] |
| rnn_cell.py | 143 forget_bias=1.0,
166 forget_bias: Biases of the forget gate are initialized by default to 1
193 self._forget_bias = forget_bias
343 forget_bias=1.0,
358 forget_bias: float, Biases of the forget gate are initialized by default
374 self._forget_bias = forget_bias
522 forget_bias=1.0,
545 forget_bias: (optional) float, default 1.0, The initial bias of the
583 self._forget_bias = forget_bias
[all...] |
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| rnn_cells.py | 45 We add forget_bias (default: 1) to the biases of the forget gate in order to
57 forget_bias=1.0,
66 forget_bias: float, The bias added to forget gates (see above).
85 forget_bias=forget_bias,
189 forget_bias=1.0,
211 forget_bias: Biases of the forget gate are initialized by default to 1
235 forget_bias=forget_bias,
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| bidirectional_sequence_lstm_test.py | 60 self.num_units, use_peepholes=True, forget_bias=0, name="rnn1"),
62 self.num_units, num_proj=8, forget_bias=0, name="rnn2"),
67 forget_bias=0,
70 self.num_units, forget_bias=0, name="rnn4")
|
| unidirectional_sequence_lstm_test.py | 59 self.num_units, use_peepholes=True, forget_bias=1.0, name="rnn1"),
61 self.num_units, num_proj=8, forget_bias=1.0, name="rnn2"),
66 forget_bias=0,
69 self.num_units, forget_bias=1.0, name="rnn4")
|
| rnn_cell.py | 199 forget_bias=1.0,
223 forget_bias: Biases of the forget gate are initialized by default to 1 in
267 self._forget_bias = forget_bias
351 self.forget_bias = add_variable_wrapped("forget_bias", bias_shape,
462 transpose_b=True), self.forget_bias)
522 "forget_bias": self._forget_bias,
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| lstm_ops_gpu.cu.cc | 89 const float forget_bias, const float cell_clip,
94 T forget_bias_t = strict_cast<T>(forget_bias);
223 OpKernelContext* ctx, const GPUDevice& d, const float forget_bias,
269 i.data(), f.data(), forget_bias, cell_clip, batch_size, cell_size));
275 i.data(), f.data(), forget_bias, cell_clip, batch_size, cell_size));
411 OpKernelContext* ctx, const GPUDevice& d, const float forget_bias, \
423 LSTMBlockCellFpropWithCUDA<T>(ctx, d, forget_bias, cell_clip, \
|
| lstm_ops.cc | 47 const float forget_bias, const float cell_clip, bool use_peephole,
91 f.constant(T(forget_bias)) + f_peep)
95 f.constant(T(forget_bias)))
184 OpKernelContext* ctx, const CPUDevice& d, const float forget_bias, \
197 *this, ctx, d, forget_bias, cell_clip, use_peephole, x, cs_prev, \
238 OP_REQUIRES_OK(ctx, ctx->GetAttr("forget_bias", &forget_bias_));
386 OpKernelContext* ctx, const GPUDevice& d, const float forget_bias, \
[all...] |
| lstm_ops.h | 157 const float forget_bias, const float cell_clip,
|
| /external/tensorflow/tensorflow/python/grappler/ |
| hierarchical_controller.py | 741 forget_bias = variable_scope.get_variable("encoder_forget_bias")
746 forget_bias = variable_scope.get_variable("encoder_forget_bias")
764 forget_bias)
769 forget_bias)
804 next_c, next_h = lstm(signal, prev_c, prev_h, w_lstm, forget_bias)
[all...] |
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| strip_pruning_vars_test.py | 114 dim, forget_bias=0.0, state_is_tuple=True, reuse=False)
|
| /external/tensorflow/tensorflow/python/ops/ |
| rnn_cell_impl.py | 640 We add forget_bias (default: 1) to the biases of the forget gate in order to
659 forget_bias=1.0,
670 forget_bias: float, The bias added to forget gates (see above).
707 self._forget_bias = forget_bias
792 "forget_bias": self._forget_bias,
[all...] |
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| rnn_cell_test.py | 209 forget_bias=1.0,
385 num_units=num_units, forget_bias=1.0, state_is_tuple=False)(x, m)
417 forget_bias=1.0)(x, m)
453 forget_bias=1.0,
501 forget_bias=1.0,
576 forget_bias=1.0,
664 forget_bias=1.0,
[all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| cudnn_rnn_ops_test.py | 126 # canonical lstm. must set forget_bias to 0. to align with cudnn lstm.
127 cell = rnn_cell_impl.LSTMCell(num_units, forget_bias=0., reuse=True)
[all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| cudnn_rnn_ops.py | 71 num_units, forget_bias=0, cell_clip=None, use_peephole=False,
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| rnn_cell_test.py | [all...] |
| /external/tensorflow/tensorflow/lite/testing/ |
| generate_examples.py | [all...] |
