MCPcopy Create free account
hub / github.com/apache/tvm / _shape_product

Function _shape_product

python/tvm/backend/cuda/lang/alloc_pool.py:65–66  ·  view source on GitHub ↗
(shape)

Source from the content-addressed store, hash-verified

63
64
65def _shape_product(shape):
66 return functools.reduce(operator.mul, shape, 1)
67
68
69def _auto_swizzle_mode(dtype):

Callers 2

_resolve_colsMethod · 0.70
allocMethod · 0.70

Calls 1

reduceMethod · 0.80

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…