Home
last modified time | relevance | path

Searched refs:input_sparse (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/libopus/dnn/torch/weight-exchange/wexchange/torch/
H A Dtorch.py180 def dump_torch_gru_weights(where, gru, name='gru', input_sparse=False, recurrent_sparse=False, quan… argument
197 …_gru_layer(where, name, w_ih, w_hh, b_ih, b_hh, format='torch', input_sparse=input_sparse, recurre…
207 def dump_torch_grucell_weights(where, gru, name='gru', input_sparse=False, recurrent_sparse=False, … argument
221 …_gru_layer(where, name, w_ih, w_hh, b_ih, b_hh, format='torch', input_sparse=input_sparse, recurre…
/aosp_15_r20/external/libopus/dnn/torch/weight-exchange/wexchange/tf/
H A Dtf.py37 def dump_tf_gru_weights(where, gru, name='gru', input_sparse=False, recurrent_sparse=False, quantiz… argument
50 …int_gru_layer(where, name, w_ih, w_hh, b_ih, b_hh, format='tf', input_sparse=input_sparse, recurre…
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dtesting_utils.py575 input_sparse=None, argument
596 if input_ragged or input_sparse:
601 sparse=input_sparse)
616 sparse=input_sparse))
629 sparse=input_sparse)
/aosp_15_r20/external/libopus/dnn/torch/osce/
H A Dexport_model_weights.py64 …('feature_net.gru', dict(quantize=True, scale=None, recurrent_scale=None, input_sparse=sparse_defa…
85 …('feature_net.gru', dict(quantize=True, scale=None, recurrent_scale=None, input_sparse=sparse_defa…
/aosp_15_r20/external/libopus/dnn/torch/rdovae/
H A Dexport_rdovae_weights.py209 …hts(enc_writer, model.get_submodule(name), export_name, verbose=True, input_sparse=True, quantize=…
252 …hts(dec_writer, model.get_submodule(name), export_name, verbose=True, input_sparse=True, quantize=…
/aosp_15_r20/external/libopus/dnn/torch/weight-exchange/wexchange/c_export/
H A Dcommon.py336 input_sparse : bool = False,
357 …print_linear_layer(writer, name + "_input", weight, bias, scale=scale, sparse=input_sparse, quanti…
/aosp_15_r20/external/libopus/dnn/torch/lossgen/
H A Dexport_lossgen.py78 …weights(writer, model.get_submodule(name), export_name, verbose=True, input_sparse=False, quantize…
/aosp_15_r20/external/libopus/dnn/torch/plc/
H A Dexport_plc.py78 …weights(writer, model.get_submodule(name), export_name, verbose=True, input_sparse=False, quantize…
/aosp_15_r20/external/libopus/dnn/torch/neural-pitch/
H A Dexport_neuralpitch_weights.py89 …weights(writer, model.get_submodule(name), export_name, verbose=True, input_sparse=False, quantize…