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

Method repeat

tensorrt_llm/functional.py:522–526  ·  view source on GitHub ↗

See functional.repeat

(self, sizes)

Source from the content-addressed store, hash-verified

520 return unbind(self, dim)
521
522 def repeat(self, sizes):
523 '''
524 See functional.repeat
525 '''
526 return repeat(self, sizes)
527
528 def is_dynamic(self, dim=None):
529 '''

Callers 15

build_alibi_tensorFunction · 0.80
test_gpt_attentionMethod · 0.80
_woq_groupwise_matmulMethod · 0.80
test_repeatMethod · 0.80
eyeMethod · 0.80
create_lora_weightsMethod · 0.80
test_rg_lruMethod · 0.80
_full_test_multi_gpuFunction · 0.80
set_tensor_value_2Function · 0.80

Calls 1

repeatFunction · 0.85

Tested by 15

test_gpt_attentionMethod · 0.64
_woq_groupwise_matmulMethod · 0.64
test_repeatMethod · 0.64
eyeMethod · 0.64
create_lora_weightsMethod · 0.64
test_rg_lruMethod · 0.64
_full_test_multi_gpuFunction · 0.64
set_tensor_value_2Function · 0.64
set_tensor_value_3Function · 0.64