Home
last modified time | relevance | path

Searched defs:gru (Results 1 – 17 of 17) sorted by relevance

/external/webrtc/modules/audio_processing/agc2/rnn_vad/
Drnn_gru_unittest.cc30 GatedRecurrentLayer& gru, in TestGatedRecurrentLayer()
113 GatedRecurrentLayer gru(kGruInputSize, kGruOutputSize, kGruBias, kGruWeights, in TEST_P() local
132 GatedRecurrentLayer gru(kInputLayerOutputSize, kHiddenLayerOutputSize, in TEST_P() local
/external/libopus/dnn/torch/weight-exchange/wexchange/tf/
Dtf.py37 def dump_tf_gru_weights(where, gru, name='gru', input_sparse=False, recurrent_sparse=False, quantiz… argument
67 def load_tf_gru_weights(path, gru): argument
/external/libopus/dnn/torch/weight-exchange/wexchange/torch/
Dtorch.py180 def dump_torch_gru_weights(where, gru, name='gru', input_sparse=False, recurrent_sparse=False, quan… argument
207 def dump_torch_grucell_weights(where, gru, name='gru', input_sparse=False, recurrent_sparse=False, … argument
232 def load_torch_gru_weights(where, gru): argument
/external/libopus/dnn/torch/lpcnet/utils/sparsification/
Dcommon.py91 def calculate_gru_flops_per_step(gru, sparsification_dict=dict(), drop_input=False): argument
Dgru_sparsifier.py174 gru = torch.nn.GRU(10, 20) variable
/external/libopus/dnn/torch/dnntools/dnntools/sparsification/
Dcommon.py93 def calculate_gru_flops_per_step(gru, sparsification_dict=dict(), drop_input=False): argument
Dgru_sparsifier.py163 gru = torch.nn.GRU(10, 20) variable
/external/libopus/dnn/torch/osce/utils/
Dendoscopy.py16 def get_gru_gates(gru, input, state): argument
/external/libopus/dnn/torch/lpcnet/utils/
Dendoscopy.py45 def get_gru_gates(gru, input, state): argument
/external/libopus/dnn/torch/osce/stndrd/presentation/
Dendoscopy.py16 def get_gru_gates(gru, input, state): argument
/external/rnnoise/src/
Drnn.c109 void compute_gru(const GRULayer *gru, float *state, const float *input) in compute_gru()
/external/libopus/src/
Dmlp.c92 void analysis_compute_gru(const AnalysisGRULayer *gru, float *state, const float *input) in analysis_compute_gru()
/external/pytorch/test/cpp/api/
Drnn.cpp399 GRU gru(GRUOptions(100, 256).num_layers(2).bidirectional(true)); in TEST_F() local
Dmodules.cpp4527 auto gru = GRUCell(1, 2); in TEST_F() local
/external/pytorch/aten/src/ATen/native/miopen/
DRNN_miopen.cpp882 ONE_HIDDEN_RNN(gru, miopenGRU) in ONE_HIDDEN_RNN() argument
/external/pytorch/aten/src/ATen/native/cudnn/
DRNN.cpp2686 ONE_HIDDEN_RNN(gru, CUDNN_GRU) in ONE_HIDDEN_RNN() argument
/external/pytorch/aten/src/ATen/native/
DRNN.cpp1376 ONE_HIDDEN_RNN(gru, GRUCell<CellParams>) in ONE_HIDDEN_RNN() argument