MCPcopy Index your code
hub / github.com/NVIDIA/TensorRT-LLM / transpose

Method transpose

tensorrt_llm/functional.py:426–430  ·  view source on GitHub ↗

See functional.transpose.

(self, dim0, dim1)

Source from the content-addressed store, hash-verified

424 return permute(self, dims)
425
426 def transpose(self, dim0, dim1):
427 '''
428 See functional.transpose.
429 '''
430 return transpose(self, dim0, dim1)
431
432 def mean(self, dim, keepdim=False):
433 '''

Callers 15

topk_forwardFunction · 0.80
sum_bitmatrix_rowsFunction · 0.80
downcast_to_mxfpFunction · 0.80
upcast_from_mxfpFunction · 0.80
downcast_to_mxfp_torchFunction · 0.80
upcast_from_mxfp_torchFunction · 0.80
swizzle_dataMethod · 0.80
swizzle_dataMethod · 0.80
unswizzle_dataMethod · 0.80
ref.pyFile · 0.80
test_gpt_attentionMethod · 0.80

Calls 1

transposeFunction · 0.85

Tested by 15

test_gpt_attentionMethod · 0.64
_construct_executionMethod · 0.64
_attnMethod · 0.64
test_fp4_gemmMethod · 0.64
test_transposeMethod · 0.64
set_weight_layerMethod · 0.64
test_ranksMethod · 0.64
test_nonzeroMethod · 0.64
_matmulMethod · 0.64
_make_latent_cache_genFunction · 0.64