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
↓ 4 callers
Method
__init__
( self, d_model=512, nhead=8, cnn_module_kernel=15, num_encoder_layers
athena/layers/conformer.py:51
↓ 4 callers
Method
__init__
( self, d_model=512, nhead=8, num_encoder_layers=6, num_decoder_layers
athena/layers/transformer.py:49
↓ 4 callers
Function
_cast_to_type_if_compatible
Cast hparam to the provided type, if compatible. Args: name: Name of the hparam to be cast. param_type: The type of the hparam.
athena/utils/hparam.py:139
↓ 4 callers
Method
_feedforward_decoder
feed-forward decoder Args: encoder_output: encoder outputs, shape: [batch, x_steps, d_model] duration_indexes: argmax
athena/models/tts/fastspeech.py:231
↓ 4 callers
Function
_parse_fail
Helper function for raising a value error for bad assignment.
athena/utils/hparam.py:47
↓ 4 callers
Function
_reuse_fail
Helper function for raising a value error for reuse of name.
athena/utils/hparam.py:55
↓ 4 callers
Function
_update_insert_pos
(old_pos, pylist)
examples/tts/data_baker/local/prepare_data.py:57
↓ 4 callers
Function
abs
runtime/core/cppflow/raw_ops.h:26
↓ 4 callers
Method
clip_by_norm
clip norm using tf.clip_by_norm
athena/solver.py:599
↓ 4 callers
Method
compute_logit_length
used for get logit length
athena/models/asr/speech_conformer_ctc.py:137
↓ 4 callers
Function
floor
runtime/core/cppflow/raw_ops.h:9507
↓ 4 callers
Function
generate_square_subsequent_mask_u2
Generate a square mask for the sequence. The masked positions are filled with bool(True). Unmasked positions are filled with bool(False).
athena/utils/misc.py:204
↓ 4 callers
Method
get_tensor
NOTE: Changes to the returned TF_Tensor are not reflected in the actual device memory!
runtime/core/cppflow/tensor.h:248
↓ 4 callers
Method
get_tensor
NOTE: Changes to the returned TF_Tensor are not reflected in the actual device memory!
runtime/server/x86/cppflow/tensor.h:248
↓ 4 callers
Function
hz2bark
Hertz to Bark conversion */
athena/transform/feats/ops/kernels/support_functions.cc:151
↓ 4 callers
Method
operator()
runtime/core/cppflow/model.h:149
↓ 4 callers
Method
operator()
runtime/server/x86/cppflow/model.h:149
↓ 4 callers
Function
parse_jsonfile
parse the jsonfile, output the parameters
athena/main.py:137
↓ 4 callers
Method
reset_states
reset num_err and num_total to zero
athena/metrics.py:44
↓ 4 callers
Function
select_dict_by_frames_num
(num_frames, input_dict)
examples/vad/google_dataset_v2/local/prepare_data.py:95
↓ 4 callers
Function
to_string
runtime/server/x86/cppflow/ops.h:81
↓ 4 callers
Method
total_score
runtime/core/decoder/ctc_prefix_beam_search.h:32
↓ 4 callers
Method
total_score
runtime/server/x86/decoder/ctc_prefix_beam_search.h:32
↓ 4 callers
Function
validate_seqs
Discard end symbol and elements after end symbol Args: seqs: shape=(batch_size, seq_length) eos: eos id Returns: val
athena/utils/misc.py:399
↓ 4 callers
Function
vector_Min
athena/transform/feats/ops/kernels/pitch.cc:23
↓ 4 callers
Method
viterbi_score
runtime/core/decoder/ctc_prefix_beam_search.h:27
↓ 4 callers
Method
viterbi_score
runtime/server/x86/decoder/ctc_prefix_beam_search.h:27
↓ 3 callers
Method
AcceptWaveform
athena/transform/feats/ops/kernels/pitch.cc:806
↓ 3 callers
Method
AcceptWaveform
runtime/core/frontend/feature_pipeline.cc:37
↓ 3 callers
Method
AcceptWaveform
runtime/server/x86/frontend/feature_pipeline.cc:37
↓ 3 callers
Method
Clear
runtime/core/utils/hash_list_inl.h:43
↓ 3 callers
Method
Clear
runtime/server/x86/utils/hash_list_inl.h:43
↓ 3 callers
Method
Find
runtime/core/utils/hash_list_inl.h:69
↓ 3 callers
Method
Find
runtime/server/x86/utils/hash_list_inl.h:69
↓ 3 callers
Method
GetAM
runtime/core/inference/resource.h:62
↓ 3 callers
Method
GetAM
runtime/server/x86/inference/resource.h:62
↓ 3 callers
Method
GetAmNnet
runtime/core/inference/tensor_model.h:81
↓ 3 callers
Method
GetAmNnet
runtime/server/x86/inference/tensor_model.h:81
↓ 3 callers
Method
GetDecodeOptions
runtime/core/inference/resource.h:68
↓ 3 callers
Method
GetDecodeOptions
runtime/server/x86/inference/resource.h:68
↓ 3 callers
Method
GetLM
runtime/core/inference/resource.h:65
↓ 3 callers
Method
GetLM
runtime/server/x86/inference/resource.h:65
↓ 3 callers
Method
Init
runtime/core/inference/e2e_context.cc:21
↓ 3 callers
Method
Init
runtime/server/x86/inference/e2e_context.cc:21
↓ 3 callers
Method
InitOptions
runtime/core/inference/resource.cc:24
↓ 3 callers
Method
InitOptions
runtime/server/x86/inference/resource.cc:24
↓ 3 callers
Method
Insert
runtime/core/utils/hash_list_inl.h:123
↓ 3 callers
Method
Insert
runtime/server/x86/utils/hash_list_inl.h:123
↓ 3 callers
Method
NccfWindowShift
Returns the window-shift in samples, after resampling.
athena/transform/feats/ops/kernels/pitch.h:114
↓ 3 callers
Function
NewDecoderResourcePtr
runtime/core/inference/interface.cc:83
↓ 3 callers
Method
RandomState
athena/transform/feats/ops/kernels/support_functions.cc:712
↓ 3 callers
Method
SetInputFinished
runtime/core/frontend/feature_pipeline.cc:57
↓ 3 callers
Method
SetInputFinished
runtime/server/x86/frontend/feature_pipeline.cc:57
↓ 3 callers
Method
__init__
(self, data_descriptions, config=None)
athena/models/vad/vad_marblenet.py:232
↓ 3 callers
Method
_encoder_forward_chunk
Forward just one chunk Args: xs (tf.Tensor), shape=[B(1), T, feat_dim, 1]: chunk input offset (int): current offset
athena/models/asr/speech_u2.py:196
↓ 3 callers
Method
_forward_encoder
(self, speech: tf.Tensor, speech_length: tf.Tensor,
athena/models/asr/speech_u2.py:160
↓ 3 callers
Method
_forward_encoder
(self, samples, training: bool = None)
athena/models/asr/av_conformer.py:201
↓ 3 callers
Method
addTask
Add a task
runtime/core/utils/thread_pool.h:122
↓ 3 callers
Function
complex_abs2
complex number absolute value square
athena/transform/feats/ops/kernels/complex_defines.h:80
↓ 3 callers
Method
decode
(self, wav_dir=None)
athena/run_demo.py:92
↓ 3 callers
Method
decode
convert a list of ids to a sentence
athena/data/text_featurizer.py:87
↓ 3 callers
Method
dtype
runtime/core/cppflow/tensor.h:243
↓ 3 callers
Method
dtype
runtime/server/x86/cppflow/tensor.h:243
↓ 3 callers
Function
exit
runtime/server/x86/cppflow/raw_ops.h:7640
↓ 3 callers
Function
get_dict_from_scp
(vocab, func=lambda x: int(x[0]))
athena/utils/misc.py:514
↓ 3 callers
Method
get_operation_shape
runtime/core/cppflow/model.h:107
↓ 3 callers
Method
get_operation_shape
runtime/server/x86/cppflow/model.h:107
↓ 3 callers
Function
hz2mel
Hertz to Mel conversion */
athena/transform/feats/ops/kernels/support_functions.cc:165
↓ 3 callers
Function
log
runtime/server/x86/cppflow/raw_ops.h:11783
↓ 3 callers
Function
nth_element
runtime/core/cppflow/raw_ops.h:14444
↓ 3 callers
Function
nth_element
runtime/server/x86/cppflow/raw_ops.h:14444
↓ 3 callers
Function
prepare_scp
(out_dir, data_type, speech_wav_list, background_wav_list)
examples/vad/google_dataset_v2/local/prepare_data.py:143
↓ 3 callers
Function
read_csv_dict
(csv_path)
athena/utils/misc.py:539
↓ 3 callers
Method
restore_from_best
restore from the best model
athena/utils/checkpoint.py:196
↓ 3 callers
Method
set_snip_edges
athena/transform/feats/ops/kernels/spectrum.cc:60
↓ 3 callers
Method
shard
creates a Dataset that includes only 1/num_shards of this dataset
athena/data/datasets/base.py:171
↓ 3 callers
Method
split_heads
Split the last dimension into (num_heads, depth). Transpose the result such that the shape is (batch_size, num_heads, seq_len, depth)
athena/layers/attention.py:139
↓ 3 callers
Method
split_heads
Split the last dimension into (num_heads, depth). Transpose the result such that the shape is (batch_size, num_heads, seq_len, depth)
athena/layers/u2/attention_u2.py:138
↓ 3 callers
Function
sqrt
runtime/server/x86/cppflow/raw_ops.h:22467
↓ 3 callers
Method
storage_features_offline
(self)
athena/data/datasets/asr/speech_recognition.py:129
↓ 3 callers
Function
to_string
runtime/core/cppflow/ops.h:81
↓ 3 callers
Method
update_state
Accumulate errors and counts
athena/metrics.py:73
↓ 2 callers
Method
ComputeBacktraces
athena/transform/feats/ops/kernels/pitch.cc:228
↓ 2 callers
Method
ComputeLatency
athena/transform/feats/ops/kernels/pitch.cc:405
↓ 2 callers
Function
ComputeNccf
athena/transform/feats/ops/kernels/pitch.cc:101
↓ 2 callers
Method
Delete
runtime/core/utils/hash_list_inl.h:63
↓ 2 callers
Method
Delete
runtime/server/x86/utils/hash_list_inl.h:63
↓ 2 callers
Method
Elapsed
return int in milliseconds
runtime/core/utils/timer.h:27
↓ 2 callers
Method
Encode
runtime/core/decoder/ctc_decodable.cc:39
↓ 2 callers
Method
Encode
runtime/server/x86/decoder/ctc_decodable.cc:39
↓ 2 callers
Function
Gcd
return gcd */
athena/transform/feats/ops/kernels/support_functions.cc:659
↓ 2 callers
Method
GetCache1Shape
runtime/core/inference/tensor_model.h:93
↓ 2 callers
Method
GetCache1Shape
runtime/server/x86/inference/tensor_model.h:93
↓ 2 callers
Method
GetCache1Size
runtime/core/inference/tensor_model.h:101
↓ 2 callers
Method
GetCache1Size
runtime/server/x86/inference/tensor_model.h:101
↓ 2 callers
Method
GetCache2Shape
runtime/core/inference/tensor_model.h:97
↓ 2 callers
Method
GetCache2Shape
runtime/server/x86/inference/tensor_model.h:97
↓ 2 callers
Method
GetCache2Size
runtime/core/inference/tensor_model.h:105
↓ 2 callers
Method
GetCache2Size
runtime/server/x86/inference/tensor_model.h:105
↓ 2 callers
Method
GetFrame
athena/transform/feats/ops/kernels/pitch.cc:698
← previous
next →
101–200 of 3,518, ranked by callers