Home
last modified time | relevance | path

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

/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/tf_lstm_ops/xsmm_lstm/
H A D__init__.py126 forget_bias=1.0, argument
272 forget_bias=1.0, argument
H A Dtest.py47 forget_bias=1.0 variable
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/tf_lstm_ops/
H A Dlstm_fwd.c56 const float forget_bias, in lstm_fwd_create()
234 const float forget_bias, in lstm_fwd_set_ptr()
/dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/
H A Dinitializer.py706 def __init__(self, forget_bias=1.0): argument
738 def __init__(self, init, num_hidden, num_layers, mode, bidirectional=False, forget_bias=1.0): argument
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/
H A Dinitializer.py706 def __init__(self, forget_bias=1.0): argument
738 def __init__(self, init, num_hidden, num_layers, mode, bidirectional=False, forget_bias=1.0): argument
/dports/math/libxsmm/libxsmm-1.16.3/src/
H A Dlibxsmm_dnn_rnncell.c1969 …rr_t libxsmm_dnn_rnncell_allocate_forget_bias(libxsmm_dnn_rnncell* handle, const float forget_bias) in libxsmm_dnn_rnncell_allocate_forget_bias()
H A Dlibxsmm_main.h693 float forget_bias; member
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/lstmdriver/
H A Dlstmdriver_nc_ck_f32.c51 float forget_bias = 1.0f; in main() local
H A Dlstmdriver_nc_ck_bf16.c56 float forget_bias = 1.0f; in main() local
H A Dlstmdriver_nc_kcck_bf16.c56 float forget_bias = 1.0f; in main() local
H A Dlstmdriver_nc_kcck_f32.c49 float forget_bias = 1.0f; in main() local
/dports/misc/tvm/incubator-tvm-0.6.1/nnvm/tests/python/frontend/tensorflow/
H A Dtest_forward.py699 def _test_lstm_cell(batch_size, num_hidden, num_layers, forget_bias, dtype): argument
/dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/tests/python/frontend/tensorflow/
H A Dtest_forward.py699 def _test_lstm_cell(batch_size, num_hidden, num_layers, forget_bias, dtype): argument
/dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/rnn/
H A Drnn_cell.py421 def __init__(self, num_hidden, prefix='lstm_', params=None, forget_bias=1.0): argument
564 dropout=0., get_next_state=False, forget_bias=1.0, argument
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/rnn/
H A Drnn_cell.py421 def __init__(self, num_hidden, prefix='lstm_', params=None, forget_bias=1.0): argument
564 dropout=0., get_next_state=False, forget_bias=1.0, argument
/dports/misc/mnn/MNN-1.2.0/source/geometry/
H A DGeometryLSTM.cpp696 float forget_bias = param->forget_bias(); in onCompute() local
/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/frontend/tensorflow/
H A Dtest_forward.py1600 def _test_lstm_cell(batch_size, num_hidden, num_layers, forget_bias, dtype): argument
/dports/misc/tvm/incubator-tvm-0.6.1/tests/python/frontend/tensorflow/
H A Dtest_forward.py1600 def _test_lstm_cell(batch_size, num_hidden, num_layers, forget_bias, dtype): argument
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/common/
H A Ddnn_common.h1249 …NLINE void matrix_copy_forget_bias(int m, int n, int ld, float *src, float *dst, float forget_bias) in matrix_copy_forget_bias()
2481 … int K, float *bigold, float *bcgold, float *bfgold, float *bogold, float forget_bias, float *icfo… in lstm_fwd_copy_bias()
2730 LIBXSMM_INLINE void lstm_ref_fwd( int N, int C, int K, int t, float forget_bias, in lstm_ref_fwd()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/frontend/tensorflow/
H A Dtest_forward.py2518 def _test_lstm_cell(batch_size, num_hidden, num_layers, forget_bias, dtype): argument
/dports/misc/mnn/MNN-1.2.0/schema/current/
H A DTensorflowOp_generated.h3321 float forget_bias; member
3338 float forget_bias() const { in forget_bias() function