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

Method size

triton_kernels/tensor.py:163–166  ·  view source on GitHub ↗
(self, i=None)

Source from the content-addressed store, hash-verified

161 return self.ndim
162
163 def size(self, i=None):
164 if i is None:
165 return self.shape
166 return self.shape[i]
167
168
169@dataclass

Callers 15

assert_closeFunction · 0.45
downcast_to_mxfp_torchFunction · 0.45
upcast_from_mxfp_torchFunction · 0.45
unswizzle_dataMethod · 0.45
postprocess_imageMethod · 0.45
print_outputFunction · 0.45
mainFunction · 0.45
mainFunction · 0.45
__call__Method · 0.45
eval_trt_llmFunction · 0.45
eval_hfFunction · 0.45
report_load_statistics.pyFile · 0.45

Calls

no outgoing calls

Tested by 1

assert_closeFunction · 0.36