| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | quant_lstm_sup_test.cc | 261 std::vector<int32_t> forget_bias; in TEST() local
263 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
284 std::vector<int32_t> forget_bias; in TEST() local
286 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
307 std::vector<int32_t> forget_bias; in TEST() local
309 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
312 EXPECT_THAT(forget_bias, ElementsAreArray({9206, -46884, -11693, -38724})); in TEST()
330 std::vector<int32_t> forget_bias; in TEST() local
332 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
|
| D | quant_lstm_sup.cc | 133 std::vector<int32_t>* forget_bias, in DecomposeBiasTensor() argument
141 forget_bias->resize(bias_size); in DecomposeBiasTensor()
143 forget_bias->begin()); in DecomposeBiasTensor()
|
| D | quant_lstm_sup.h | 51 std::vector<int32_t>* forget_bias,
|
| D | nnapi_delegate.cc | 3300 std::vector<int32_t> forget_bias; in Map() local
3304 &forget_bias, &output_bias); in Map()
3313 forget_bias, biases_tensor.params, &forget_bias_tensor); in Map()
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | unidirectional_sequence_lstm_test.py | 61 self.num_units, use_peepholes=True, forget_bias=1.0, name="rnn1"),
63 self.num_units, num_proj=8, forget_bias=1.0, name="rnn2"),
68 forget_bias=0,
71 self.num_units, forget_bias=1.0, name="rnn4")
|
| D | bidirectional_sequence_lstm_test.py | 61 self.num_units, use_peepholes=True, forget_bias=0, name="rnn1"),
63 self.num_units, num_proj=8, forget_bias=0, name="rnn2"),
68 forget_bias=0,
71 self.num_units, forget_bias=0, name="rnn4")
|
| D | rnn_cell.py | 206 forget_bias=1.0, argument
274 self._forget_bias = forget_bias
363 self.forget_bias = add_variable_wrapped("forget_bias", bias_shape,
474 transpose_b=True), self.forget_bias)
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops_gpu.cu.cc | 88 const float forget_bias, const float cell_clip, in lstm_gates() argument
93 T forget_bias_t = strict_cast<T>(forget_bias); in lstm_gates()
228 OpKernelContext* ctx, const GPUDevice& d, const float forget_bias, in LSTMBlockCellFpropWithCUDA() argument
274 co.data(), i.data(), f.data(), forget_bias, cell_clip, batch_size, in LSTMBlockCellFpropWithCUDA()
281 co.data(), i.data(), f.data(), forget_bias, cell_clip, batch_size, in LSTMBlockCellFpropWithCUDA()
413 OpKernelContext* ctx, const GPUDevice& d, const float forget_bias, \
426 ctx, d, forget_bias, cell_clip, use_peephole, x, cs_prev, h_prev, w, \
|
| D | lstm_ops.cc | 47 const float forget_bias, const float cell_clip, bool use_peephole, in LSTMBlockCellFpropWithEigen() argument
94 f.constant(T(forget_bias)) + f_peep) in LSTMBlockCellFpropWithEigen()
99 f.constant(T(forget_bias))) in LSTMBlockCellFpropWithEigen()
191 OpKernelContext* ctx, const CPUDevice& d, const float forget_bias, \
204 *this, ctx, d, forget_bias, cell_clip, use_peephole, x, cs_prev, \
251 OpKernelContext* ctx, const GPUDevice& d, const float forget_bias, \
|
| D | lstm_ops.h | 169 const float forget_bias, const float cell_clip,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_LSTMBlockCell.pbtxt | 95 name: "forget_bias"
122 f = f + forget_bias
|
| D | api_def_BlockLSTM.pbtxt | 102 name: "forget_bias"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_grad_test.py | 53 forget_bias=0.0, # Disable to match V2 default.
96 forget_bias=1.0,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | LSTMBlockCell.pbtxt | 64 name: "forget_bias"
|
| D | BlockLSTM.pbtxt | 68 name: "forget_bias"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | LSTMBlockCell.pbtxt | 64 name: "forget_bias"
|
| D | BlockLSTM.pbtxt | 68 name: "forget_bias"
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | lstm.py | 62 num_cells, forget_bias=0.0, state_is_tuple=True)
|
| /external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ |
| D | rnn_cell_impl.py | 678 forget_bias=1.0, argument
729 self._forget_bias = forget_bias
863 forget_bias=1.0, argument
942 self._forget_bias = forget_bias
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
| D | TensorFlow_NumPy_Text_Generation.ipynb | 491 " def __init__(self, n_units, forget_bias=0.0):\n",
493 " self._forget_bias = forget_bias\n",
530 " def __init__(self, n_units, forget_bias=0.0, stateful=False):\n",
531 " self._cell = GRUCell(n_units, forget_bias)\n",
605 …" def __init__(self, vocab_size, embedding_dim, rnn_units, forget_bias=0.0, stateful=False, activ…
607 " self._gru = GRU(rnn_units, forget_bias=forget_bias, stateful=stateful)\n",
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.rnn_cell.-l-s-t-m-cell.pbtxt | 151 …num_proj\', \'proj_clip\', \'num_unit_shards\', \'num_proj_shards\', \'forget_bias\', \'state_is_t…
|
| D | tensorflow.nn.rnn_cell.-basic-l-s-t-m-cell.pbtxt | 151 …argspec: "args=[\'self\', \'num_units\', \'forget_bias\', \'state_is_tuple\', \'activation\', \'re…
|
| D | tensorflow.lite.experimental.nn.-t-f-lite-l-s-t-m-cell.pbtxt | 151 …num_proj\', \'proj_clip\', \'num_unit_shards\', \'num_proj_shards\', \'forget_bias\', \'state_is_t…
|
| D | tensorflow.raw_ops.pbtxt | 521 …', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\', \'b\', \'forget_bias\', \'cell_clip\…
2081 …', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\', \'b\', \'forget_bias\', \'cell_clip\…
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 2764 forget_bias=1.0,
2800 forget_bias=1.0,
|