MCPcopy
hub / github.com/NVIDIA/TensorRT-LLM / sqrt

Method sqrt

tensorrt_llm/functional.py:450–454  ·  view source on GitHub ↗

See functional.sqrt.

(self)

Source from the content-addressed store, hash-verified

448 return abs(self)
449
450 def sqrt(self):
451 '''
452 See functional.sqrt.
453 '''
454 return sqrt(self)
455
456 def squeeze(self, dim, zero_is_placeholder):
457 '''

Callers 15

assert_closeFunction · 0.80
run_testFunction · 0.80
mha_refFunction · 0.80
mha_refFunction · 0.80
geluMethod · 0.80
create_weightsMethod · 0.80
create_lora_weightsMethod · 0.80
calculate_ref_resultFunction · 0.80
calculate_ref_result_ctxFunction · 0.80
calculate_ref_result_genFunction · 0.80
test_flashinfer_prefillFunction · 0.80
test_attention_backendFunction · 0.80

Calls

no outgoing calls

Tested by 15

assert_closeFunction · 0.64
run_testFunction · 0.64
geluMethod · 0.64
create_weightsMethod · 0.64
create_lora_weightsMethod · 0.64
calculate_ref_resultFunction · 0.64
calculate_ref_result_ctxFunction · 0.64
calculate_ref_result_genFunction · 0.64
test_flashinfer_prefillFunction · 0.64
test_attention_backendFunction · 0.64
test_attention_backend_ifbFunction · 0.64
test_flash_mla_sparse_fwdFunction · 0.64