MCPcopy Create free account

hub / github.com/athena-team/athena / functions

Functions3,518 in github.com/athena-team/athena

Methodforward
(self, inputs, inputs_length=None, training: bool = None)
athena/models/lm/rnn_lm.py:64
Methodforward
(self, inputs, inputs_length=None, training: bool = None, ignored_id=[])
athena/models/lm/kenlm.py:31
Methodforward
(self, inputs, input_lengths, training: bool = None)
athena/models/lm/transformer_lm.py:61
Methodforward_chunk_by_chunk
Forward input chunk by chunk with chunk_size like a streaming fashion Here we should pay special attention to computation cache
athena/layers/u2/conformer_u2.py:275
Functionfractional_avg_pool_grad
runtime/core/cppflow/raw_ops.h:9587
Functionfractional_avg_pool_grad
runtime/server/x86/cppflow/raw_ops.h:9587
Functionfractional_max_pool_grad
runtime/core/cppflow/raw_ops.h:9623
Functionfractional_max_pool_grad
runtime/server/x86/cppflow/raw_ops.h:9623
Functionfreeze2tflite
(jsonfile, freezed_model_dir)
athena/freeze.py:100
Methodfreeze_beam_search
beam search for freeze only support batch=1 Args: samples: the data source to be decoded beam_size: beam size
athena/models/asr/speech_transformer.py:305
Methodfreeze_beam_search
beam search for freeze only support batch=1 Args: samples: the data source to be decoded beam_size: beam size
athena/models/asr/speech_conformer.py:345
Methodfreeze_beam_search
beam search for freeze only support batch=1 Args: samples: the data source to be decoded beam_size: beam size
athena/models/asr/speech_conformer_ctc.py:259
Functionfreeze_ctc_prefix_beam_decoder
CTC prefix beam search inner implementation reference: https://github.com/wenet-e2e/wenet/blob/main/wenet/transformer/asr_model.py. A
athena/tools/ctc_decoder.py:105
Methodfreeze_ctc_prefix_beam_search
( self, samples, ctc_final_layer, hparams=None,beam_size=1 )
athena/models/asr/speech_transformer.py:543
Methodfreeze_ctc_prefix_beam_search
( self, samples, ctc_final_layer, hparams=None,beam_size=1 )
athena/models/asr/speech_conformer_ctc.py:223
Functionfreeze_e2e
entry point for model inference, do some preparation work
athena/freeze.py:37
Functionfreeze_tts
freeze TensorFlow model trained on Python to saved model,
athena/freeze.py:86
Functionfreeze_u2
entry point for model inference, do some preparation work
athena/freeze.py:191
Functionfresnel_cos
runtime/core/cppflow/raw_ops.h:9663
Functionfresnel_cos
runtime/server/x86/cppflow/raw_ops.h:9663
Functionfresnel_sin
runtime/core/cppflow/raw_ops.h:9687
Functionfresnel_sin
runtime/server/x86/cppflow/raw_ops.h:9687
Functionfused_pad_conv2_d
runtime/core/cppflow/raw_ops.h:9711
Functionfused_pad_conv2_d
runtime/server/x86/cppflow/raw_ops.h:9711
Functionfused_resize_and_pad_conv2_d
runtime/core/cppflow/raw_ops.h:9745
Functionfused_resize_and_pad_conv2_d
runtime/server/x86/cppflow/raw_ops.h:9745
Functiongather
runtime/core/cppflow/raw_ops.h:9784
Functiongather
runtime/server/x86/cppflow/raw_ops.h:9784
Functiongather_nd
runtime/core/cppflow/raw_ops.h:9814
Functiongather_nd
runtime/server/x86/cppflow/raw_ops.h:9814
Functiongather_v2
runtime/core/cppflow/raw_ops.h:9843
Functiongather_v2
runtime/server/x86/cppflow/raw_ops.h:9843
Functiongen_dctmatrix
generate DCT matrix */
athena/transform/feats/ops/kernels/support_functions.cc:247
Functiongen_warpweights
generate frequency scale warping */
athena/transform/feats/ops/kernels/support_functions.cc:179
Functiongenerator
(i)
athena/utils/data_queue.py:92
Methodgenerator_task
Enqueue batch data
athena/utils/data_queue.py:63
Methodget_config
(self)
athena/layers/commons.py:168
Methodget_context
runtime/server/x86/cppflow/context.h:56
Methodget_data
runtime/core/cppflow/tensor.h:217
Methodget_data
runtime/server/x86/cppflow/tensor.h:217
Methodget_eager_handle
TODO: Need to determine if we can mark the return value or *this as const
runtime/core/cppflow/tensor.h:99
Methodget_eager_handle
TODO: Need to determine if we can mark the return value or *this as const
runtime/server/x86/cppflow/tensor.h:99
Methodget_init
(self)
athena/models/asr/mtl_seq2seq.py:183
Methodget_loss
get loss
athena/models/base.py:37
Methodget_loss
get MPC loss Args: logits: MPC output Return:: MPC L1 loss and metrics
athena/models/masked_pc.py:138
Methodget_loss
get loss used for training
athena/models/asr/mtl_seq2seq.py:79
Methodget_loss
get loss used for training
athena/models/asr/av_mtl_seq2seq.py:70
Methodget_loss
(self, outputs, samples, training=None)
athena/models/vad/vad_dnn.py:75
Methodget_loss
(self, outputs, samples, training=None)
athena/models/vad/vad_marblenet.py:308
Methodget_loss
get loss used for training
athena/models/tts/fastspeech.py:223
Methodget_loss
(self, outputs, samples, training=None)
athena/models/tts/tacotron2.py:432
Methodget_loss
get loss
athena/models/kws/base.py:37
Methodget_loss
get loss
athena/models/kws/transformer_wakeup_focal_loss.py:151
Methodget_model_structure
(self)
athena/utils/hparam.py:542
Methodget_operations
runtime/core/cppflow/model.h:95
Methodget_operations
runtime/server/x86/cppflow/model.h:95
Methodget_ppl
loss is returned by self.get_loss(), and it is a tensor with none shape float value
athena/models/lm/nn_lm.py:66
Functionget_session_handle
runtime/core/cppflow/raw_ops.h:9878
Functionget_session_handle
runtime/server/x86/cppflow/raw_ops.h:9878
Functionget_session_handle_v2
runtime/core/cppflow/raw_ops.h:9902
Functionget_session_handle_v2
runtime/server/x86/cppflow/raw_ops.h:9902
Functionget_session_tensor
runtime/core/cppflow/raw_ops.h:9926
Functionget_session_tensor
runtime/server/x86/cppflow/raw_ops.h:9926
Methodget_status
runtime/core/cppflow/context.h:60
Methodget_status
runtime/server/x86/cppflow/context.h:60
Methodget_subsample_rate
(self)
athena/models/asr/mtl_seq2seq.py:179
Functionglu
Gated Linear Unit. Original paper: https://arxiv.org/abs/1612.08083 Args: x: float Tensor to perform activation. Returns: `x
athena/layers/functional.py:115
Functiongreater
runtime/core/cppflow/raw_ops.h:9950
Functiongreater
runtime/server/x86/cppflow/raw_ops.h:9950
Functiongreater_equal
runtime/core/cppflow/raw_ops.h:9978
Functiongreater_equal
runtime/server/x86/cppflow/raw_ops.h:9978
Functionguarantee_const_tensor
runtime/core/cppflow/raw_ops.h:10006
Functionguarantee_const_tensor
runtime/server/x86/cppflow/raw_ops.h:10006
Functionh_s_v_to_r_g_b
runtime/core/cppflow/raw_ops.h:10030
Functionh_s_v_to_r_g_b
runtime/server/x86/cppflow/raw_ops.h:10030
Functionhash_table
runtime/core/cppflow/raw_ops.h:10054
Functionhash_table
runtime/server/x86/cppflow/raw_ops.h:10054
Functionhash_table_v2
runtime/core/cppflow/raw_ops.h:10079
Functionhash_table_v2
runtime/server/x86/cppflow/raw_ops.h:10079
Functionhistogram_fixed_width
runtime/core/cppflow/raw_ops.h:10104
Functionhistogram_fixed_width
runtime/server/x86/cppflow/raw_ops.h:10104
Functionhistogram_summary
runtime/core/cppflow/raw_ops.h:10136
Functionhistogram_summary
runtime/server/x86/cppflow/raw_ops.h:10136
Functioni_f_f_t
runtime/core/cppflow/raw_ops.h:10164
Functioni_f_f_t
runtime/server/x86/cppflow/raw_ops.h:10164
Functioni_f_f_t2_d
runtime/core/cppflow/raw_ops.h:10188
Functioni_f_f_t2_d
runtime/server/x86/cppflow/raw_ops.h:10188
Functioni_f_f_t3_d
runtime/core/cppflow/raw_ops.h:10212
Functioni_f_f_t3_d
runtime/server/x86/cppflow/raw_ops.h:10212
Functioni_r_f_f_t
runtime/core/cppflow/raw_ops.h:10236
Functioni_r_f_f_t
runtime/server/x86/cppflow/raw_ops.h:10236
Functioni_r_f_f_t2_d
runtime/core/cppflow/raw_ops.h:10265
Functioni_r_f_f_t2_d
runtime/server/x86/cppflow/raw_ops.h:10265
Functioni_r_f_f_t3_d
runtime/core/cppflow/raw_ops.h:10294
Functioni_r_f_f_t3_d
runtime/server/x86/cppflow/raw_ops.h:10294
Functionidentity
runtime/core/cppflow/raw_ops.h:10323
Functionidentity
runtime/server/x86/cppflow/raw_ops.h:10323
Functionidentity_n
runtime/core/cppflow/raw_ops.h:10347
Functionidentity_n
runtime/server/x86/cppflow/raw_ops.h:10347
Functionidentity_reader
runtime/core/cppflow/raw_ops.h:10373
← previousnext →2,001–2,100 of 3,518, ranked by callers