MCPcopy Create free account

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

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

↓ 4 callersMethod__init__
( self, d_model=512, nhead=8, cnn_module_kernel=15, num_encoder_layers
athena/layers/conformer.py:51
↓ 4 callersMethod__init__
( self, d_model=512, nhead=8, num_encoder_layers=6, num_decoder_layers
athena/layers/transformer.py:49
↓ 4 callersFunction_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 callersMethod_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 callersFunction_parse_fail
Helper function for raising a value error for bad assignment.
athena/utils/hparam.py:47
↓ 4 callersFunction_reuse_fail
Helper function for raising a value error for reuse of name.
athena/utils/hparam.py:55
↓ 4 callersFunction_update_insert_pos
(old_pos, pylist)
examples/tts/data_baker/local/prepare_data.py:57
↓ 4 callersFunctionabs
runtime/core/cppflow/raw_ops.h:26
↓ 4 callersMethodclip_by_norm
clip norm using tf.clip_by_norm
athena/solver.py:599
↓ 4 callersMethodcompute_logit_length
used for get logit length
athena/models/asr/speech_conformer_ctc.py:137
↓ 4 callersFunctionfloor
runtime/core/cppflow/raw_ops.h:9507
↓ 4 callersFunctiongenerate_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 callersMethodget_tensor
NOTE: Changes to the returned TF_Tensor are not reflected in the actual device memory!
runtime/core/cppflow/tensor.h:248
↓ 4 callersMethodget_tensor
NOTE: Changes to the returned TF_Tensor are not reflected in the actual device memory!
runtime/server/x86/cppflow/tensor.h:248
↓ 4 callersFunctionhz2bark
Hertz to Bark conversion */
athena/transform/feats/ops/kernels/support_functions.cc:151
↓ 4 callersMethodoperator()
runtime/core/cppflow/model.h:149
↓ 4 callersMethodoperator()
runtime/server/x86/cppflow/model.h:149
↓ 4 callersFunctionparse_jsonfile
parse the jsonfile, output the parameters
athena/main.py:137
↓ 4 callersMethodreset_states
reset num_err and num_total to zero
athena/metrics.py:44
↓ 4 callersFunctionselect_dict_by_frames_num
(num_frames, input_dict)
examples/vad/google_dataset_v2/local/prepare_data.py:95
↓ 4 callersFunctionto_string
runtime/server/x86/cppflow/ops.h:81
↓ 4 callersMethodtotal_score
runtime/core/decoder/ctc_prefix_beam_search.h:32
↓ 4 callersMethodtotal_score
runtime/server/x86/decoder/ctc_prefix_beam_search.h:32
↓ 4 callersFunctionvalidate_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 callersFunctionvector_Min
athena/transform/feats/ops/kernels/pitch.cc:23
↓ 4 callersMethodviterbi_score
runtime/core/decoder/ctc_prefix_beam_search.h:27
↓ 4 callersMethodviterbi_score
runtime/server/x86/decoder/ctc_prefix_beam_search.h:27
↓ 3 callersMethodAcceptWaveform
athena/transform/feats/ops/kernels/pitch.cc:806
↓ 3 callersMethodAcceptWaveform
runtime/core/frontend/feature_pipeline.cc:37
↓ 3 callersMethodAcceptWaveform
runtime/server/x86/frontend/feature_pipeline.cc:37
↓ 3 callersMethodClear
runtime/core/utils/hash_list_inl.h:43
↓ 3 callersMethodClear
runtime/server/x86/utils/hash_list_inl.h:43
↓ 3 callersMethodFind
runtime/core/utils/hash_list_inl.h:69
↓ 3 callersMethodFind
runtime/server/x86/utils/hash_list_inl.h:69
↓ 3 callersMethodGetAM
runtime/core/inference/resource.h:62
↓ 3 callersMethodGetAM
runtime/server/x86/inference/resource.h:62
↓ 3 callersMethodGetAmNnet
runtime/core/inference/tensor_model.h:81
↓ 3 callersMethodGetAmNnet
runtime/server/x86/inference/tensor_model.h:81
↓ 3 callersMethodGetDecodeOptions
runtime/core/inference/resource.h:68
↓ 3 callersMethodGetDecodeOptions
runtime/server/x86/inference/resource.h:68
↓ 3 callersMethodGetLM
runtime/core/inference/resource.h:65
↓ 3 callersMethodGetLM
runtime/server/x86/inference/resource.h:65
↓ 3 callersMethodInit
runtime/core/inference/e2e_context.cc:21
↓ 3 callersMethodInit
runtime/server/x86/inference/e2e_context.cc:21
↓ 3 callersMethodInitOptions
runtime/core/inference/resource.cc:24
↓ 3 callersMethodInitOptions
runtime/server/x86/inference/resource.cc:24
↓ 3 callersMethodInsert
runtime/core/utils/hash_list_inl.h:123
↓ 3 callersMethodInsert
runtime/server/x86/utils/hash_list_inl.h:123
↓ 3 callersMethodNccfWindowShift
Returns the window-shift in samples, after resampling.
athena/transform/feats/ops/kernels/pitch.h:114
↓ 3 callersFunctionNewDecoderResourcePtr
runtime/core/inference/interface.cc:83
↓ 3 callersMethodRandomState
athena/transform/feats/ops/kernels/support_functions.cc:712
↓ 3 callersMethodSetInputFinished
runtime/core/frontend/feature_pipeline.cc:57
↓ 3 callersMethodSetInputFinished
runtime/server/x86/frontend/feature_pipeline.cc:57
↓ 3 callersMethod__init__
(self, data_descriptions, config=None)
athena/models/vad/vad_marblenet.py:232
↓ 3 callersMethod_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 callersMethod_forward_encoder
(self, speech: tf.Tensor, speech_length: tf.Tensor,
athena/models/asr/speech_u2.py:160
↓ 3 callersMethod_forward_encoder
(self, samples, training: bool = None)
athena/models/asr/av_conformer.py:201
↓ 3 callersMethodaddTask
Add a task
runtime/core/utils/thread_pool.h:122
↓ 3 callersFunctioncomplex_abs2
complex number absolute value square
athena/transform/feats/ops/kernels/complex_defines.h:80
↓ 3 callersMethoddecode
(self, wav_dir=None)
athena/run_demo.py:92
↓ 3 callersMethoddecode
convert a list of ids to a sentence
athena/data/text_featurizer.py:87
↓ 3 callersMethoddtype
runtime/core/cppflow/tensor.h:243
↓ 3 callersMethoddtype
runtime/server/x86/cppflow/tensor.h:243
↓ 3 callersFunctionexit
runtime/server/x86/cppflow/raw_ops.h:7640
↓ 3 callersFunctionget_dict_from_scp
(vocab, func=lambda x: int(x[0]))
athena/utils/misc.py:514
↓ 3 callersMethodget_operation_shape
runtime/core/cppflow/model.h:107
↓ 3 callersMethodget_operation_shape
runtime/server/x86/cppflow/model.h:107
↓ 3 callersFunctionhz2mel
Hertz to Mel conversion */
athena/transform/feats/ops/kernels/support_functions.cc:165
↓ 3 callersFunctionlog
runtime/server/x86/cppflow/raw_ops.h:11783
↓ 3 callersFunctionnth_element
runtime/core/cppflow/raw_ops.h:14444
↓ 3 callersFunctionnth_element
runtime/server/x86/cppflow/raw_ops.h:14444
↓ 3 callersFunctionprepare_scp
(out_dir, data_type, speech_wav_list, background_wav_list)
examples/vad/google_dataset_v2/local/prepare_data.py:143
↓ 3 callersFunctionread_csv_dict
(csv_path)
athena/utils/misc.py:539
↓ 3 callersMethodrestore_from_best
restore from the best model
athena/utils/checkpoint.py:196
↓ 3 callersMethodset_snip_edges
athena/transform/feats/ops/kernels/spectrum.cc:60
↓ 3 callersMethodshard
creates a Dataset that includes only 1/num_shards of this dataset
athena/data/datasets/base.py:171
↓ 3 callersMethodsplit_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 callersMethodsplit_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 callersFunctionsqrt
runtime/server/x86/cppflow/raw_ops.h:22467
↓ 3 callersMethodstorage_features_offline
(self)
athena/data/datasets/asr/speech_recognition.py:129
↓ 3 callersFunctionto_string
runtime/core/cppflow/ops.h:81
↓ 3 callersMethodupdate_state
Accumulate errors and counts
athena/metrics.py:73
↓ 2 callersMethodComputeBacktraces
athena/transform/feats/ops/kernels/pitch.cc:228
↓ 2 callersMethodComputeLatency
athena/transform/feats/ops/kernels/pitch.cc:405
↓ 2 callersFunctionComputeNccf
athena/transform/feats/ops/kernels/pitch.cc:101
↓ 2 callersMethodDelete
runtime/core/utils/hash_list_inl.h:63
↓ 2 callersMethodDelete
runtime/server/x86/utils/hash_list_inl.h:63
↓ 2 callersMethodElapsed
return int in milliseconds
runtime/core/utils/timer.h:27
↓ 2 callersMethodEncode
runtime/core/decoder/ctc_decodable.cc:39
↓ 2 callersMethodEncode
runtime/server/x86/decoder/ctc_decodable.cc:39
↓ 2 callersFunctionGcd
return gcd */
athena/transform/feats/ops/kernels/support_functions.cc:659
↓ 2 callersMethodGetCache1Shape
runtime/core/inference/tensor_model.h:93
↓ 2 callersMethodGetCache1Shape
runtime/server/x86/inference/tensor_model.h:93
↓ 2 callersMethodGetCache1Size
runtime/core/inference/tensor_model.h:101
↓ 2 callersMethodGetCache1Size
runtime/server/x86/inference/tensor_model.h:101
↓ 2 callersMethodGetCache2Shape
runtime/core/inference/tensor_model.h:97
↓ 2 callersMethodGetCache2Shape
runtime/server/x86/inference/tensor_model.h:97
↓ 2 callersMethodGetCache2Size
runtime/core/inference/tensor_model.h:105
↓ 2 callersMethodGetCache2Size
runtime/server/x86/inference/tensor_model.h:105
↓ 2 callersMethodGetFrame
athena/transform/feats/ops/kernels/pitch.cc:698
← previousnext →101–200 of 3,518, ranked by callers