MCPcopy Index your code
hub / github.com/apache/tvm / greater_equal

Method greater_equal

tests/python/relax/test_transform_legalize_ops_binary.py:1400–1407  ·  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_greater_equal: T.Buffer((T.int64(4), T.int64(3), T.int64(2), T.int64(3)), "bool"))

Source from the content-addressed store, hash-verified

1398
1399 @T.prim_func(private=True, s_tir=True)
1400 def greater_equal(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_greater_equal: T.Buffer((T.int64(4), T.int64(3), T.int64(2), T.int64(3)), "bool")):
1401 T.func_attr({"tirx.noalias": True})
1402 for i0, i1, i2, i3 in T.grid(T.int64(4), T.int64(3), T.int64(2), T.int64(3)):
1403 with T.sblock("T_greater_equal"):
1404 ax0, ax1, ax2, ax3 = T.axis.remap("SSSS", [i0, i1, i2, i3])
1405 T.reads(rxplaceholder_1[ax0, ax1, ax2, T.int64(0)], rxplaceholder[T.int64(0), ax2, ax3])
1406 T.writes(T_greater_equal[ax0, ax1, ax2, ax3])
1407 T_greater_equal[ax0, ax1, ax2, ax3] = rxplaceholder_1[ax0, ax1, ax2, T.int64(0)] <= rxplaceholder[T.int64(0), ax2, ax3]
1408 # fmt: on
1409
1410 mod = LegalizeOps()(GreaterEqual)

Callers 14

greater_equalFunction · 0.80
__ge__Method · 0.80
_impl_v13Method · 0.80
greater_equalFunction · 0.80
slice_scatterFunction · 0.80
greater_equalFunction · 0.80
multibox_transform_locFunction · 0.80
mainMethod · 0.80
mainMethod · 0.80
test_op_correctnessFunction · 0.80
mainMethod · 0.80
mainMethod · 0.80

Calls 1

remapMethod · 0.80

Tested by

no test coverage detected