Home
last modified time | relevance | path

Searched refs:prove_equal (Results 1 – 3 of 3) sorted by relevance

/dports/misc/py-tvm/incubator-tvm-0.6.1/src/op/
H A Dscan_op.cc41 inline bool prove_equal(Expr lhs, Expr rhs) { in prove_equal() function
83 CHECK(prove_equal(init[i]->shape[0], axis->dom->min)) in make()
85 CHECK(prove_equal( in make()
94 CHECK(prove_equal( in make()
108 CHECK(prove_equal( in make()
/dports/misc/tvm/incubator-tvm-0.6.1/src/op/
H A Dscan_op.cc41 inline bool prove_equal(Expr lhs, Expr rhs) { in prove_equal() function
83 CHECK(prove_equal(init[i]->shape[0], axis->dom->min)) in make()
85 CHECK(prove_equal( in make()
94 CHECK(prove_equal( in make()
108 CHECK(prove_equal( in make()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/te/operation/
H A Dscan_op.cc68 auto prove_equal = [&](PrimExpr lhs, PrimExpr rhs) { in ScanOp() local
75 CHECK(prove_equal(init[i]->shape[0], axis->dom->min)) in ScanOp()
77 CHECK(prove_equal(state_placeholder[i]->shape[0], axis->dom->min + axis->dom->extent)) in ScanOp()
85 CHECK(prove_equal(update[i]->shape[k], state_placeholder[i]->shape[k])); in ScanOp()
96 CHECK(prove_equal(init[i]->shape[k], state_placeholder[i]->shape[k])); in ScanOp()