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

Method minimum

tests/python/relax/test_transform_legalize_ops_binary.py:2101–2108  ·  view source on GitHub ↗
(rxplaceholder: T.Buffer((T.int64(1), T.int64(2), T.int64(3)), "float32"), rxplaceholder_1: T.Buffer((T.int64(4), T.int64(3), T.int64(2), T.int64(1)), "float32"), T_minimum: T.Buffer((T.int64(4), T.int64(3), T.int64(2), T.int64(3)), "float32"))

Source from the content-addressed store, hash-verified

2099
2100 @T.prim_func(private=True, s_tir=True)
2101 def minimum(rxplaceholder: T.Buffer((T.int64(1), T.int64(2), T.int64(3)), "float32"), rxplaceholder_1: T.Buffer((T.int64(4), T.int64(3), T.int64(2), T.int64(1)), "float32"), T_minimum: T.Buffer((T.int64(4), T.int64(3), T.int64(2), T.int64(3)), "float32")):
2102 T.func_attr({"tirx.noalias": True})
2103 for i0, i1, i2, i3 in T.grid(T.int64(4), T.int64(3), T.int64(2), T.int64(3)):
2104 with T.sblock("T_minimum"):
2105 ax0, ax1, ax2, ax3 = T.axis.remap("SSSS", [i0, i1, i2, i3])
2106 T.reads(rxplaceholder[T.int64(0), ax2, ax3], rxplaceholder_1[ax0, ax1, ax2, T.int64(0)])
2107 T.writes(T_minimum[ax0, ax1, ax2, ax3])
2108 T_minimum[ax0, ax1, ax2, ax3] = T.min(rxplaceholder[T.int64(0), ax2, ax3], rxplaceholder_1[ax0, ax1, ax2, T.int64(0)])
2109 # fmt: on
2110
2111 mod = LegalizeOps()(Minimum)

Callers 15

mainMethod · 0.45
mainMethod · 0.45
mainMethod · 0.45
mainMethod · 0.45
mainMethod · 0.45
fnFunction · 0.45
mainMethod · 0.45
mainMethod · 0.45
apply_attentionFunction · 0.45
testMethod · 0.45
testFunction · 0.45
test_minimumMethod · 0.45

Calls 2

remapMethod · 0.80
minMethod · 0.80

Tested by

no test coverage detected