MCPcopy Create free account

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

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

↓ 2 callersMethodGetResult
runtime/core/inference/e2e_context.h:39
↓ 2 callersMethodInitDecoding
runtime/core/decoder/ctc_faster_decoder.cc:36
↓ 2 callersMethodInitDecoding
runtime/server/x86/decoder/ctc_faster_decoder.cc:36
↓ 2 callersMethodInputFinished
athena/transform/feats/ops/kernels/pitch.cc:705
↓ 2 callersMethodIsBlankFrame
runtime/core/decoder/ctc_decodable.cc:69
↓ 2 callersMethodIsBlankFrame
runtime/server/x86/decoder/ctc_decodable.cc:69
↓ 2 callersMethodIsCTCPrune
runtime/core/decoder/ctc_decodable.h:76
↓ 2 callersMethodIsCTCPrune
runtime/server/x86/decoder/ctc_decodable.h:76
↓ 2 callersMethodLoadWords
runtime/core/inference/tensor_model.h:177
↓ 2 callersMethodLoadWords
runtime/server/x86/inference/tensor_model.h:177
↓ 2 callersFunctionLog
athena/transform/feats/ops/kernels/support_functions.h:153
↓ 2 callersMethodLogLikelihood
runtime/core/decoder/ctc_decodable.cc:81
↓ 2 callersMethodLogLikelihood
runtime/server/x86/decoder/ctc_decodable.cc:81
↓ 2 callersMethodNumFramesReady
athena/transform/feats/ops/kernels/pitch.cc:690
↓ 2 callersFunctionRand
athena/transform/feats/ops/kernels/support_functions.cc:698
↓ 2 callersFunctionRandUniform
Returns a random number strictly between 0 and 1.
athena/transform/feats/ops/kernels/support_functions.h:156
↓ 2 callersMethodSetBestState
athena/transform/feats/ops/kernels/pitch.cc:381
↓ 2 callersMethodSetSize
runtime/core/utils/hash_list_inl.h:34
↓ 2 callersMethodSetSize
runtime/server/x86/utils/hash_list_inl.h:34
↓ 2 callersMethod__init__
( self, d_model=512, nhead=8, cnn_module_kernel=15, num_encoder_layers
athena/layers/conformer_ctc.py:46
↓ 2 callersMethod__init__
( self, d_model=512, nhead=8, num_encoder_layers=6,
athena/layers/u2/transformer_u2.py:52
↓ 2 callersMethod__init__
( self, d_model=512, nhead=8, cnn_module_kernel=15,
athena/layers/u2/conformer_u2.py:53
↓ 2 callersMethod__init__
(self, data_descriptions, config=None)
athena/models/asr/speech_u2.py:58
↓ 2 callersMethod__init__
(self, data_descriptions, config=None)
athena/models/tts/fastspeech2.py:62
↓ 2 callersMethod__init__
(self, data_descriptions, config=None)
athena/models/tts/fastspeech.py:63
↓ 2 callersMethod__str__
(self)
athena/utils/hparam.py:637
↓ 2 callersMethod_apply_weightnorm
Try apply weightnorm for all layer in list_layers.
athena/models/vad/vad_marblenet.py:166
↓ 2 callersMethod_forward_encoder
(self, speech, speech_length, training: bool = None)
athena/models/asr/speech_transformer.py:167
↓ 2 callersMethod_forward_encoder
(self, speech, speech_length, training: bool = None)
athena/models/asr/speech_conformer.py:164
↓ 2 callersMethod_remove_outlier
Remove outlier from x.
examples/tts/data_baker/local/extract_feat.py:42
↓ 2 callersMethod_synthesize_post_net
Args: before_outs: the outputs before postnet logits_stack: the logits of all steps Returns:: af
athena/models/tts/tacotron2.py:514
↓ 2 callersMethodaddTask
Add a task
runtime/server/x86/utils/thread_pool.h:122
↓ 2 callersFunctionadd_optional_chunk_mask_u2
Apply optional mask for encoder. Args: xs (tf.Tensor): padded input, (B, L, D), L for max length mask (tf.Tensor): mask for xs,
athena/utils/misc.py:282
↓ 2 callersMethodas_dataset
return tf.data.Dataset object
athena/data/datasets/asr/speech_recognition_batch_bins.py:193
↓ 2 callersMethodattention_rescoring
Apply attention rescoring decoding, CTC prefix beam search is applied first to get nbest, then we resoring the nbest on atten
athena/models/asr/speech_u2.py:381
↓ 2 callersMethodbeam_search
batch beam search for transformer model Args: samples: the data source to be decoded beam_size: beam size
athena/models/asr/speech_u2.py:569
↓ 2 callersMethodbuild
build in keras layer
athena/layers/attention.py:250
↓ 2 callersMethodcall
(self, x)
athena/layers/commons.py:81
↓ 2 callersFunctioncollapse4d
reshape from [N T D C] -> [N T D*C] using tf.shape(x), which generate a tensor instead of x.shape
athena/layers/functional.py:41
↓ 2 callersFunctioncombine_dict_to_csv
(phone_dur_dict, feat_dict, total_csv_path)
examples/tts/data_baker/local/prepare_data.py:327
↓ 2 callersFunctioncomplex_add
complex addition
athena/transform/feats/ops/kernels/complex_defines.h:119
↓ 2 callersMethodcompute_cmvn_if_necessary
compute cmvn file
athena/data/datasets/base.py:208
↓ 2 callersMethodcompute_cmvn_kaldiio
compute cmvn for filtered entries using kaldi-format data
athena/data/feature_normalizer.py:146
↓ 2 callersMethodcompute_fs2_cmvn
compuate cmvn of mel-spec,f0 and energy
athena/data/feature_normalizer.py:232
↓ 2 callersMethodcompute_logit_length
compute the logit length
athena/models/base.py:51
↓ 2 callersMethodcompute_logit_length
compute the logit length
athena/models/asr/mtl_seq2seq.py:96
↓ 2 callersMethodcompute_logit_length
used for get logit length
athena/models/asr/av_conformer.py:192
↓ 2 callersFunctioncompute_mean
compute mean */
athena/transform/feats/ops/kernels/support_functions.cc:33
↓ 2 callersMethodcompute_score
Args: value_length: the length of value, shape: [batch] max_len: the maximun length Returns: init
athena/layers/attention.py:391
↓ 2 callersMethodevaluate
evaluate the model
athena/solver.py:130
↓ 2 callersMethodevaluate
evaluate the model
athena/solver.py:667
↓ 2 callersFunctionexp
runtime/server/x86/cppflow/raw_ops.h:7664
↓ 2 callersFunctionextract_feat
(wav_list, label, start = 0.0, end = None)
examples/vad/google_dataset_v2/local/prepare_data.py:65
↓ 2 callersMethodget_config
(self)
athena/layers/commons.py:480
↓ 2 callersMethodget_encoder_init_input
(self)
athena/models/asr/speech_u2.py:329
↓ 2 callersMethodinference
synthesize using vocoder on dataset
athena/run_demo.py:56
↓ 2 callersMethodinitialize_input_y
Args: y: the true label, shape: [batch, y_steps, feat_dim] Returns:: y0: zeros will be padded as one step to
athena/models/tts/tacotron2.py:338
↓ 2 callersMethodinitialize_states
Args: encoder_output: encoder outputs, shape: [batch, x_step, eunits] input_length: shape: [batch] Returns::
athena/models/tts/tacotron2.py:352
↓ 2 callersFunctioninsert_eos_in_labels
insert eos in labels, batch size should be larger than 1 assuming 0 as the padding,
athena/utils/misc.py:51
↓ 2 callersMethodjoin
runtime/server/x86/utils/thread_pool.h:57
↓ 2 callersMethodload_model
load model
athena/data/text_featurizer.py:179
↓ 2 callersFunctionmake_positional_encoding
generate a postional encoding list
athena/layers/functional.py:25
↓ 2 callersFunctionmel2hz
Mel to Hertz conversion */
athena/transform/feats/ops/kernels/support_functions.cc:172
↓ 2 callersMethodnum_channels
(self)
athena/transform/feats/fbank.py:172
↓ 2 callersMethodparse_json
Override existing hyperparameter values, parsing new values from a json object. Args: values_json: String containing a json objec
athena/utils/hparam.py:580
↓ 2 callersFunctionparse_name
runtime/core/cppflow/model.h:144
↓ 2 callersFunctionparse_name
runtime/server/x86/cppflow/model.h:144
↓ 2 callersFunctionpow
runtime/server/x86/cppflow/raw_ops.h:15499
↓ 2 callersMethodread_shape_file
(self, file_dir=None)
athena/data/datasets/asr/speech_recognition_batch_bins_kaldiio.py:150
↓ 2 callersFunctionreverse
runtime/core/cppflow/raw_ops.h:18298
↓ 2 callersFunctionreverse
runtime/server/x86/cppflow/raw_ops.h:18298
↓ 2 callersMethodrun
runtime/core/bin/decoder_main.cc:43
↓ 2 callersMethodscore
runtime/core/decoder/ctc_prefix_beam_search.h:26
↓ 2 callersMethodscore
runtime/server/x86/decoder/ctc_prefix_beam_search.h:26
↓ 2 callersMethodset_frame_length_sec
athena/transform/feats/ops/kernels/spectrum.cc:54
↓ 2 callersMethodset_remove_dc_offset
athena/transform/feats/ops/kernels/spectrum.cc:66
↓ 2 callersMethodset_window_length_sec
athena/transform/feats/ops/kernels/spectrum.cc:50
↓ 2 callersFunctionsplit_train_dev_test
(total_csv, output_dir)
examples/tts/data_baker/local/prepare_data.py:365
↓ 2 callersFunctionsplit_train_val_test
(dataset_dir, dev_size=0.1, test_size=0.1)
examples/vad/google_dataset_v2/local/prepare_data.py:43
↓ 2 callersMethodstart
runtime/core/utils/thread_pool.h:49
↓ 2 callersMethodstop
(self)
athena/utils/data_queue.py:60
↓ 2 callersFunctionsubsequent_chunk_mask
Create mask for subsequent steps (size, size) with chunk size, this is for streaming encoder Args: size (int): size of mask
athena/utils/misc.py:94
↓ 2 callersFunctionsubsequent_chunk_mask_u2
Create mask for subsequent steps (size, size) with chunk size, this is for streaming encoder Args: size (int): size of mask
athena/utils/misc.py:239
↓ 2 callersFunctiontf_average_by_duration
(x, durs)
athena/data/datasets/tts/speech_fastspeech2.py:43
↓ 2 callersFunctiontf_index_select
input_(tensor): input tensor dim(int): dimension indices(list): selected indices list
athena/tools/ctc_scorer.py:23
↓ 2 callersMethodtime_propagate
Args: encoder_output: encoder output (batch, x_steps, eunits). input_length: (batch,) prev_y: one step of
athena/models/tts/tacotron2.py:392
↓ 2 callersFunctiontrain
entry point for model training, implements train loop :param jsonfile: json file to read configuration from :param Solver: an abstract c
athena/main.py:183
↓ 2 callersFunctionwrite_csv
(csv_path, data: list)
athena/utils/misc.py:560
↓ 1 callersMethodAddArc
runtime/core/fst/athena_fst.h:119
↓ 1 callersMethodAddArc
runtime/server/x86/fst/athena_fst.h:119
↓ 1 callersMethodAddState
runtime/core/fst/athena_fst.h:115
↓ 1 callersMethodAddState
runtime/server/x86/fst/athena_fst.h:115
↓ 1 callersMethodAdvanceDecoding
runtime/core/decoder/ctc_faster_decoder.cc:60
↓ 1 callersMethodAdvanceDecoding
runtime/server/x86/decoder/ctc_faster_decoder.cc:60
↓ 1 callersFunctionApproxEqual
athena/transform/feats/ops/kernels/support_functions.h:134
↓ 1 callersMethodAt
runtime/core/frontend/recycling_vector.cc:33
↓ 1 callersMethodAt
runtime/server/x86/frontend/recycling_vector.cc:33
↓ 1 callersMethodClear
runtime/core/frontend/recycling_vector.cc:58
↓ 1 callersMethodClear
runtime/server/x86/frontend/recycling_vector.cc:58
↓ 1 callersMethodCompute
Compute fbank feat, return num frames
runtime/core/frontend/fbank.h:130
← previousnext →201–300 of 3,518, ranked by callers