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

Method mean

tensorrt_llm/functional.py:432–436  ·  view source on GitHub ↗

See functional.mean.

(self, dim, keepdim=False)

Source from the content-addressed store, hash-verified

430 return transpose(self, dim0, dim1)
431
432 def mean(self, dim, keepdim=False):
433 '''
434 See functional.mean.
435 '''
436 return mean(self, dim, keepdim)
437
438 def max(self, dim, keepdim=False):
439 '''

Callers 15

assert_closeFunction · 0.80
matmul_launch_metadataFunction · 0.80
maeFunction · 0.80
parse_perf_dataFunction · 0.80
check_accuracyFunction · 0.80
rms_normFunction · 0.80
rms_normFunction · 0.80
compare_outputsMethod · 0.80
rms_normFunction · 0.80
_run_test_for_backendFunction · 0.80

Calls 1

meanFunction · 0.85

Tested by 15

assert_closeFunction · 0.64
check_accuracyFunction · 0.64
rms_normFunction · 0.64
rms_normFunction · 0.64
compare_outputsMethod · 0.64
rms_normFunction · 0.64
_run_test_for_backendFunction · 0.64
_run_test_for_backendFunction · 0.64
allcloseFunction · 0.64
test_flashinfer_prefillFunction · 0.64