Home
last modified time | relevance | path

Searched defs:forget_bias (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dquant_lstm_sup_test.cc261 std::vector<int32_t> forget_bias; in TEST() local
284 std::vector<int32_t> forget_bias; in TEST() local
307 std::vector<int32_t> forget_bias; in TEST() local
330 std::vector<int32_t> forget_bias; in TEST() local
Dquant_lstm_sup.cc133 std::vector<int32_t>* forget_bias, in DecomposeBiasTensor()
Dnnapi_delegate.cc4129 std::vector<int32_t> forget_bias; in Map() local
/external/tensorflow/tensorflow/core/kernels/rnn/
Dlstm_ops_gpu.cu.cc88 const float forget_bias, const float cell_clip, in lstm_gates()
228 OpKernelContext* ctx, const GPUDevice& d, const float forget_bias, in LSTMBlockCellFpropWithCUDA()
Dlstm_ops.cc47 const float forget_bias, const float cell_clip, bool use_peephole, in LSTMBlockCellFpropWithEigen()
/external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/
Drnn_cell_impl.py681 forget_bias=1.0, argument
867 forget_bias=1.0, argument