MCPcopy
hub / github.com/apache/tvm / get_numpy

Function get_numpy

tests/python/contrib/test_cblas.py:45–50  ·  view source on GitHub ↗
(a, b, matrix_bias, transa, transb)

Source from the content-addressed store, hash-verified

43 )
44
45 def get_numpy(a, b, matrix_bias, transa, transb):
46 if transa:
47 a = a.transpose()
48 if transb:
49 b = b.transpose()
50 return np.dot(a, b) + matrix_bias
51
52 def compiling(f, name="test_matmul_add", ext=".so"):
53 path = name + ext

Callers 1

verifyFunction · 0.70

Calls 2

batch_matmulMethod · 0.80
transposeMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…