Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/athena-team/athena
/ functions
Functions
3,518 in github.com/athena-team/athena
⨍
Functions
3,518
◇
Types & classes
359
↳
Endpoints
1
↓ 2 callers
Method
GetResult
runtime/core/inference/e2e_context.h:39
↓ 2 callers
Method
InitDecoding
runtime/core/decoder/ctc_faster_decoder.cc:36
↓ 2 callers
Method
InitDecoding
runtime/server/x86/decoder/ctc_faster_decoder.cc:36
↓ 2 callers
Method
InputFinished
athena/transform/feats/ops/kernels/pitch.cc:705
↓ 2 callers
Method
IsBlankFrame
runtime/core/decoder/ctc_decodable.cc:69
↓ 2 callers
Method
IsBlankFrame
runtime/server/x86/decoder/ctc_decodable.cc:69
↓ 2 callers
Method
IsCTCPrune
runtime/core/decoder/ctc_decodable.h:76
↓ 2 callers
Method
IsCTCPrune
runtime/server/x86/decoder/ctc_decodable.h:76
↓ 2 callers
Method
LoadWords
runtime/core/inference/tensor_model.h:177
↓ 2 callers
Method
LoadWords
runtime/server/x86/inference/tensor_model.h:177
↓ 2 callers
Function
Log
athena/transform/feats/ops/kernels/support_functions.h:153
↓ 2 callers
Method
LogLikelihood
runtime/core/decoder/ctc_decodable.cc:81
↓ 2 callers
Method
LogLikelihood
runtime/server/x86/decoder/ctc_decodable.cc:81
↓ 2 callers
Method
NumFramesReady
athena/transform/feats/ops/kernels/pitch.cc:690
↓ 2 callers
Function
Rand
athena/transform/feats/ops/kernels/support_functions.cc:698
↓ 2 callers
Function
RandUniform
Returns a random number strictly between 0 and 1.
athena/transform/feats/ops/kernels/support_functions.h:156
↓ 2 callers
Method
SetBestState
athena/transform/feats/ops/kernels/pitch.cc:381
↓ 2 callers
Method
SetSize
runtime/core/utils/hash_list_inl.h:34
↓ 2 callers
Method
SetSize
runtime/server/x86/utils/hash_list_inl.h:34
↓ 2 callers
Method
__init__
( self, d_model=512, nhead=8, cnn_module_kernel=15, num_encoder_layers
athena/layers/conformer_ctc.py:46
↓ 2 callers
Method
__init__
( self, d_model=512, nhead=8, num_encoder_layers=6,
athena/layers/u2/transformer_u2.py:52
↓ 2 callers
Method
__init__
( self, d_model=512, nhead=8, cnn_module_kernel=15,
athena/layers/u2/conformer_u2.py:53
↓ 2 callers
Method
__init__
(self, data_descriptions, config=None)
athena/models/asr/speech_u2.py:58
↓ 2 callers
Method
__init__
(self, data_descriptions, config=None)
athena/models/tts/fastspeech2.py:62
↓ 2 callers
Method
__init__
(self, data_descriptions, config=None)
athena/models/tts/fastspeech.py:63
↓ 2 callers
Method
__str__
(self)
athena/utils/hparam.py:637
↓ 2 callers
Method
_apply_weightnorm
Try apply weightnorm for all layer in list_layers.
athena/models/vad/vad_marblenet.py:166
↓ 2 callers
Method
_forward_encoder
(self, speech, speech_length, training: bool = None)
athena/models/asr/speech_transformer.py:167
↓ 2 callers
Method
_forward_encoder
(self, speech, speech_length, training: bool = None)
athena/models/asr/speech_conformer.py:164
↓ 2 callers
Method
_remove_outlier
Remove outlier from x.
examples/tts/data_baker/local/extract_feat.py:42
↓ 2 callers
Method
_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 callers
Method
addTask
Add a task
runtime/server/x86/utils/thread_pool.h:122
↓ 2 callers
Function
add_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 callers
Method
as_dataset
return tf.data.Dataset object
athena/data/datasets/asr/speech_recognition_batch_bins.py:193
↓ 2 callers
Method
attention_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 callers
Method
beam_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 callers
Method
build
build in keras layer
athena/layers/attention.py:250
↓ 2 callers
Method
call
(self, x)
athena/layers/commons.py:81
↓ 2 callers
Function
collapse4d
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 callers
Function
combine_dict_to_csv
(phone_dur_dict, feat_dict, total_csv_path)
examples/tts/data_baker/local/prepare_data.py:327
↓ 2 callers
Function
complex_add
complex addition
athena/transform/feats/ops/kernels/complex_defines.h:119
↓ 2 callers
Method
compute_cmvn_if_necessary
compute cmvn file
athena/data/datasets/base.py:208
↓ 2 callers
Method
compute_cmvn_kaldiio
compute cmvn for filtered entries using kaldi-format data
athena/data/feature_normalizer.py:146
↓ 2 callers
Method
compute_fs2_cmvn
compuate cmvn of mel-spec,f0 and energy
athena/data/feature_normalizer.py:232
↓ 2 callers
Method
compute_logit_length
compute the logit length
athena/models/base.py:51
↓ 2 callers
Method
compute_logit_length
compute the logit length
athena/models/asr/mtl_seq2seq.py:96
↓ 2 callers
Method
compute_logit_length
used for get logit length
athena/models/asr/av_conformer.py:192
↓ 2 callers
Function
compute_mean
compute mean */
athena/transform/feats/ops/kernels/support_functions.cc:33
↓ 2 callers
Method
compute_score
Args: value_length: the length of value, shape: [batch] max_len: the maximun length Returns: init
athena/layers/attention.py:391
↓ 2 callers
Method
evaluate
evaluate the model
athena/solver.py:130
↓ 2 callers
Method
evaluate
evaluate the model
athena/solver.py:667
↓ 2 callers
Function
exp
runtime/server/x86/cppflow/raw_ops.h:7664
↓ 2 callers
Function
extract_feat
(wav_list, label, start = 0.0, end = None)
examples/vad/google_dataset_v2/local/prepare_data.py:65
↓ 2 callers
Method
get_config
(self)
athena/layers/commons.py:480
↓ 2 callers
Method
get_encoder_init_input
(self)
athena/models/asr/speech_u2.py:329
↓ 2 callers
Method
inference
synthesize using vocoder on dataset
athena/run_demo.py:56
↓ 2 callers
Method
initialize_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 callers
Method
initialize_states
Args: encoder_output: encoder outputs, shape: [batch, x_step, eunits] input_length: shape: [batch] Returns::
athena/models/tts/tacotron2.py:352
↓ 2 callers
Function
insert_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 callers
Method
join
runtime/server/x86/utils/thread_pool.h:57
↓ 2 callers
Method
load_model
load model
athena/data/text_featurizer.py:179
↓ 2 callers
Function
make_positional_encoding
generate a postional encoding list
athena/layers/functional.py:25
↓ 2 callers
Function
mel2hz
Mel to Hertz conversion */
athena/transform/feats/ops/kernels/support_functions.cc:172
↓ 2 callers
Method
num_channels
(self)
athena/transform/feats/fbank.py:172
↓ 2 callers
Method
parse_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 callers
Function
parse_name
runtime/core/cppflow/model.h:144
↓ 2 callers
Function
parse_name
runtime/server/x86/cppflow/model.h:144
↓ 2 callers
Function
pow
runtime/server/x86/cppflow/raw_ops.h:15499
↓ 2 callers
Method
read_shape_file
(self, file_dir=None)
athena/data/datasets/asr/speech_recognition_batch_bins_kaldiio.py:150
↓ 2 callers
Function
reverse
runtime/core/cppflow/raw_ops.h:18298
↓ 2 callers
Function
reverse
runtime/server/x86/cppflow/raw_ops.h:18298
↓ 2 callers
Method
run
runtime/core/bin/decoder_main.cc:43
↓ 2 callers
Method
score
runtime/core/decoder/ctc_prefix_beam_search.h:26
↓ 2 callers
Method
score
runtime/server/x86/decoder/ctc_prefix_beam_search.h:26
↓ 2 callers
Method
set_frame_length_sec
athena/transform/feats/ops/kernels/spectrum.cc:54
↓ 2 callers
Method
set_remove_dc_offset
athena/transform/feats/ops/kernels/spectrum.cc:66
↓ 2 callers
Method
set_window_length_sec
athena/transform/feats/ops/kernels/spectrum.cc:50
↓ 2 callers
Function
split_train_dev_test
(total_csv, output_dir)
examples/tts/data_baker/local/prepare_data.py:365
↓ 2 callers
Function
split_train_val_test
(dataset_dir, dev_size=0.1, test_size=0.1)
examples/vad/google_dataset_v2/local/prepare_data.py:43
↓ 2 callers
Method
start
runtime/core/utils/thread_pool.h:49
↓ 2 callers
Method
stop
(self)
athena/utils/data_queue.py:60
↓ 2 callers
Function
subsequent_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 callers
Function
subsequent_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 callers
Function
tf_average_by_duration
(x, durs)
athena/data/datasets/tts/speech_fastspeech2.py:43
↓ 2 callers
Function
tf_index_select
input_(tensor): input tensor dim(int): dimension indices(list): selected indices list
athena/tools/ctc_scorer.py:23
↓ 2 callers
Method
time_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 callers
Function
train
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 callers
Function
write_csv
(csv_path, data: list)
athena/utils/misc.py:560
↓ 1 callers
Method
AddArc
runtime/core/fst/athena_fst.h:119
↓ 1 callers
Method
AddArc
runtime/server/x86/fst/athena_fst.h:119
↓ 1 callers
Method
AddState
runtime/core/fst/athena_fst.h:115
↓ 1 callers
Method
AddState
runtime/server/x86/fst/athena_fst.h:115
↓ 1 callers
Method
AdvanceDecoding
runtime/core/decoder/ctc_faster_decoder.cc:60
↓ 1 callers
Method
AdvanceDecoding
runtime/server/x86/decoder/ctc_faster_decoder.cc:60
↓ 1 callers
Function
ApproxEqual
athena/transform/feats/ops/kernels/support_functions.h:134
↓ 1 callers
Method
At
runtime/core/frontend/recycling_vector.cc:33
↓ 1 callers
Method
At
runtime/server/x86/frontend/recycling_vector.cc:33
↓ 1 callers
Method
Clear
runtime/core/frontend/recycling_vector.cc:58
↓ 1 callers
Method
Clear
runtime/server/x86/frontend/recycling_vector.cc:58
↓ 1 callers
Method
Compute
Compute fbank feat, return num frames
runtime/core/frontend/fbank.h:130
← previous
next →
201–300 of 3,518, ranked by callers