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

Method squeeze

tensorrt_llm/functional.py:456–460  ·  view source on GitHub ↗

See functional.squeeze.

(self, dim, zero_is_placeholder)

Source from the content-addressed store, hash-verified

454 return sqrt(self)
455
456 def squeeze(self, dim, zero_is_placeholder):
457 '''
458 See functional.squeeze.
459 '''
460 return squeeze(self, dim, zero_is_placeholder)
461
462 def unsqueeze(self, dim):
463 '''

Callers 15

reduce_groupedFunction · 0.45
matmul_ogsFunction · 0.45
downcast_to_mxfp_torchFunction · 0.45
test_gpt_attentionMethod · 0.45
test_sage_attentionMethod · 0.45
test_squeezeMethod · 0.45
test_mlp_comparisonMethod · 0.45
test_gatherNDMethod · 0.45
test_mamba_conv1dMethod · 0.45

Calls 1

squeezeFunction · 0.85

Tested by 15

test_gpt_attentionMethod · 0.36
test_sage_attentionMethod · 0.36
test_squeezeMethod · 0.36
test_mlp_comparisonMethod · 0.36
test_gatherNDMethod · 0.36
test_mamba_conv1dMethod · 0.36
test_selective_scanMethod · 0.36
get_logprobsFunction · 0.36