MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / matmul

Function matmul

src/opr/impl/blas.cpp:616–620  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

614
615template <typename T>
616T matmul(T a, T b, const opr::BatchedMatrixMul::Param& param = {}) {
617 if (!a.node() || !b.node())
618 return {};
619 return opr::BatchedMatrixMul::make(unsafe(a), unsafe(b), param);
620}
621
622SafeSymbolVar matmuls(
623 SafeSymbolVar x, SafeSymbolVar y,

Callers 2

matmulsFunction · 0.70
MGB_IMPL_OPR_GRADFunction · 0.70

Calls 3

unsafeFunction · 0.85
makeFunction · 0.50
nodeMethod · 0.45

Tested by

no test coverage detected