Home
last modified time | relevance | path

Searched refs:relay (Results 1 – 25 of 4188) sorted by relevance

12345678910>>...168

/dports/misc/tvm/incubator-tvm-0.6.1/tests/python/relay/
H A Dtest_pass_unmatched_cases.py19 from tvm import relay
25 v = relay.Var('v')
35 v = relay.Var('v')
37 relay.Clause(relay.PatternWildcard(), v)
44 relay.Clause(relay.PatternVar(w), w)
59 relay.Clause(relay.PatternConstructor(box_ctor, [relay.PatternWildcard()]), v)
114 relay.Clause(relay.PatternWildcard(), v)
262 relay.Clause(relay.PatternConstructor(p.cons, [relay.PatternWildcard(),
265 relay.Clause(relay.PatternConstructor(p.nil, []), v)
273 clause = relay.Clause(relay.PatternTuple([relay.PatternVar(a), relay.PatternVar(b)]), a + b)
[all …]
H A Dtest_pass_alpha_equal.py275 tup = relay.Tuple([v0, relay.const(2), relay.const(3), relay.Tuple([relay.const(4)])])
276 same = relay.Tuple([v0, relay.const(2), relay.const(3), relay.Tuple([relay.const(4)])])
284 let_mapped = relay.Let(v2, relay.Tuple([v0, relay.const(2), relay.const(3),
290 … more_fields = relay.Tuple([v1, relay.const(2), relay.const(3), relay.Tuple([relay.const(4)]), v2])
428 call = relay.Call(v1, [relay.const(1), relay.const(2), v2, relay.Tuple([])],
503 if_sample = relay.If(v1, relay.const(1), relay.Tuple([relay.const(2), relay.const(3)]))
504 same = relay.If(v1, relay.const(1), relay.Tuple([relay.const(2), relay.const(3)]))
507 different_cond = relay.If(v2, relay.const(1), relay.Tuple([relay.const(2), relay.const(3)]))
510 different_true = relay.If(v1, relay.const(2), relay.Tuple([relay.const(2), relay.const(3)]))
644 x = relay.Match(relay.Tuple([relay.const(1), relay.const(1)]), [clause])
[all …]
H A Dtest_ir_parser.py269 relay.Tuple([relay.Tuple([]), relay.Tuple([]), relay.const(1)])
293 assert parses_as("(0, 1)", relay.Tuple([relay.const(0), relay.const(1)]))
295 assert parses_as("(0, 1, 2)", relay.Tuple([relay.const(0), relay.const(1), relay.const(2)]))
413 relay.multiply(relay.const(10), relay.Call(id_func, [relay.const(10)]))
460 relay.Call(multiply, [relay.const(0), relay.const(0)], None, None)
520 relay.Call(relay.op.get("abs"), [relay.const(1)], None, None)
589 relay.Function([relay.Var("x", int32)], relay.const(0), int32, []),
600 … relay.Function([relay.Var("x", int32), relay.Var("y", int32)], relay.const(0), int32, []),
624 relay.Tuple([relay.const(0)]),
635 relay.Tuple([relay.const(0), relay.const(1)]),
[all …]
H A Dtest_py_converter.py118 let = relay.Let(v, relay.Tuple([]), relay.Tuple([v, v]))
134 assign = relay.Let(v, relay.RefCreate(relay.Tuple([])), relay.RefRead(v))
142 initial_write = relay.Let(v, relay.RefCreate(relay.Tuple([relay.const(1)])),
151 v, relay.RefCreate(relay.Tuple([relay.const(1)])),
154 seq(relay.RefWrite(v, relay.Tuple([relay.const(2)])),
173 true_expr = relay.Let(v, relay.RefCreate(relay.const(0)),
175 false_expr = relay.Let(v, relay.RefCreate(relay.const(0)),
326 let = relay.Let(f, relay.Function([v], relay.Match(v, [
431 relay.Let(x, relay.Tuple([relay.const(1), relay.const(2)]),
433 relay.Let(r, relay.RefCreate(relay.const(1)),
[all …]
H A Dtest_pass_fuse_ops.py27 y = relay.add(x, relay.const(1, "float32"))
34 y = relay.add(x, relay.const(1, "float32"))
54 x = relay.add(x, relay.const(1, "float32"))
55 y = relay.nn.conv2d(x, relay.var("w1"),
60 y1 = relay.add(relay.const(1, "float32"), y)
63 z2 = relay.nn.conv2d(y, relay.var("w2"),
67 z3 = relay.nn.conv2d(y, relay.var("w3"),
78 y = relay.add(x, relay.const(1, "float32"))
119 y = relay.Call(f1, [y, relay.var("w1")])
120 z2 = relay.Call(f2, [y, relay.var("w3")])
[all …]
H A Dtest_type_infer.py42 if isinstance(expr, (relay.GlobalVar, relay.Function)):
128 f = relay.Function([x], relay.log(x))
151 with sb.if_scope(relay.equal(n, relay.const(0, ti32))):
154 sb.ret(f(relay.subtract(n, relay.const(1, ti32)), relay.log(data)))
165 func = relay.Function([x, f], relay.Call(f, [x]), tt)
178 f = relay.Var('f', relay.FuncType([b], b))
215 relay.If(f(), id_func, relay.Function([z], z)),
275 func = relay.Function([x], relay.Call(gv, [x]), tt)
288 … assert ft.checked_type == relay.FuncType([relay.scalar_type('int32')], relay.scalar_type('bool'))
319 v = relay.Var('v', relay.TensorType((), 'float32'))
[all …]
H A Dtest_pass_alter_op_layout.py20 from tvm import relay
42 y = relay.nn.relu(y)
48 weight = relay.multiply(weight, relay.const(2.0, "float32"))
54 y = relay.nn.conv2d(x, relay.multiply(weight, relay.const(2.0, "float32")),
58 y = relay.nn.relu(y)
329 y = relay.add(y, relay.const(1, "float32"))
349 y = relay.add(y, relay.const(1.0, "float32"))
576 alpha = relay.var("alpha", relay.IncompleteType())
591 alpha = relay.var("alpha", relay.IncompleteType())
952 weight = relay.multiply(weight, relay.const(2.0, "float32"))
[all …]
/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/relay/
H A Dtest_pass_unmatched_cases.py19 from tvm import relay
25 v = relay.Var('v')
35 v = relay.Var('v')
37 relay.Clause(relay.PatternWildcard(), v)
44 relay.Clause(relay.PatternVar(w), w)
59 relay.Clause(relay.PatternConstructor(box_ctor, [relay.PatternWildcard()]), v)
114 relay.Clause(relay.PatternWildcard(), v)
262 relay.Clause(relay.PatternConstructor(p.cons, [relay.PatternWildcard(),
265 relay.Clause(relay.PatternConstructor(p.nil, []), v)
273 clause = relay.Clause(relay.PatternTuple([relay.PatternVar(a), relay.PatternVar(b)]), a + b)
[all …]
H A Dtest_pass_alpha_equal.py275 tup = relay.Tuple([v0, relay.const(2), relay.const(3), relay.Tuple([relay.const(4)])])
276 same = relay.Tuple([v0, relay.const(2), relay.const(3), relay.Tuple([relay.const(4)])])
284 let_mapped = relay.Let(v2, relay.Tuple([v0, relay.const(2), relay.const(3),
290 … more_fields = relay.Tuple([v1, relay.const(2), relay.const(3), relay.Tuple([relay.const(4)]), v2])
428 call = relay.Call(v1, [relay.const(1), relay.const(2), v2, relay.Tuple([])],
503 if_sample = relay.If(v1, relay.const(1), relay.Tuple([relay.const(2), relay.const(3)]))
504 same = relay.If(v1, relay.const(1), relay.Tuple([relay.const(2), relay.const(3)]))
507 different_cond = relay.If(v2, relay.const(1), relay.Tuple([relay.const(2), relay.const(3)]))
510 different_true = relay.If(v1, relay.const(2), relay.Tuple([relay.const(2), relay.const(3)]))
644 x = relay.Match(relay.Tuple([relay.const(1), relay.const(1)]), [clause])
[all …]
H A Dtest_ir_parser.py269 relay.Tuple([relay.Tuple([]), relay.Tuple([]), relay.const(1)])
293 assert parses_as("(0, 1)", relay.Tuple([relay.const(0), relay.const(1)]))
295 assert parses_as("(0, 1, 2)", relay.Tuple([relay.const(0), relay.const(1), relay.const(2)]))
413 relay.multiply(relay.const(10), relay.Call(id_func, [relay.const(10)]))
460 relay.Call(multiply, [relay.const(0), relay.const(0)], None, None)
520 relay.Call(relay.op.get("abs"), [relay.const(1)], None, None)
589 relay.Function([relay.Var("x", int32)], relay.const(0), int32, []),
600 … relay.Function([relay.Var("x", int32), relay.Var("y", int32)], relay.const(0), int32, []),
624 relay.Tuple([relay.const(0)]),
635 relay.Tuple([relay.const(0), relay.const(1)]),
[all …]
H A Dtest_py_converter.py118 let = relay.Let(v, relay.Tuple([]), relay.Tuple([v, v]))
134 assign = relay.Let(v, relay.RefCreate(relay.Tuple([])), relay.RefRead(v))
142 initial_write = relay.Let(v, relay.RefCreate(relay.Tuple([relay.const(1)])),
151 v, relay.RefCreate(relay.Tuple([relay.const(1)])),
154 seq(relay.RefWrite(v, relay.Tuple([relay.const(2)])),
173 true_expr = relay.Let(v, relay.RefCreate(relay.const(0)),
175 false_expr = relay.Let(v, relay.RefCreate(relay.const(0)),
326 let = relay.Let(f, relay.Function([v], relay.Match(v, [
431 relay.Let(x, relay.Tuple([relay.const(1), relay.const(2)]),
433 relay.Let(r, relay.RefCreate(relay.const(1)),
[all …]
H A Dtest_pass_fuse_ops.py27 y = relay.add(x, relay.const(1, "float32"))
34 y = relay.add(x, relay.const(1, "float32"))
54 x = relay.add(x, relay.const(1, "float32"))
55 y = relay.nn.conv2d(x, relay.var("w1"),
60 y1 = relay.add(relay.const(1, "float32"), y)
63 z2 = relay.nn.conv2d(y, relay.var("w2"),
67 z3 = relay.nn.conv2d(y, relay.var("w3"),
78 y = relay.add(x, relay.const(1, "float32"))
119 y = relay.Call(f1, [y, relay.var("w1")])
120 z2 = relay.Call(f2, [y, relay.var("w3")])
[all …]
H A Dtest_type_infer.py42 if isinstance(expr, (relay.GlobalVar, relay.Function)):
128 f = relay.Function([x], relay.log(x))
151 with sb.if_scope(relay.equal(n, relay.const(0, ti32))):
154 sb.ret(f(relay.subtract(n, relay.const(1, ti32)), relay.log(data)))
165 func = relay.Function([x, f], relay.Call(f, [x]), tt)
178 f = relay.Var('f', relay.FuncType([b], b))
215 relay.If(f(), id_func, relay.Function([z], z)),
275 func = relay.Function([x], relay.Call(gv, [x]), tt)
288 … assert ft.checked_type == relay.FuncType([relay.scalar_type('int32')], relay.scalar_type('bool'))
319 v = relay.Var('v', relay.TensorType((), 'float32'))
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/relay/
H A Dtest_ir_structural_equal_hash.py344 tup = relay.Tuple([v0, relay.const(2), relay.const(3), relay.Tuple([relay.const(4)])])
345 same = relay.Tuple([v0, relay.const(2), relay.const(3), relay.Tuple([relay.const(4)])])
353 v2, relay.Tuple([v0, relay.const(2), relay.const(3), relay.Tuple([relay.const(4)])]), v2
366 different_end = relay.Tuple([v1, relay.const(2), relay.const(3), relay.Tuple([relay.const(5)])])
370 [v2, relay.const(2), relay.const(3), relay.Tuple([relay.const(4)])]
375 [v1, relay.const(2), relay.const(3), relay.Tuple([relay.const(4), relay.const(5)])]
516 [relay.const(1), relay.const(2), v2, relay.Tuple([]), relay.const(3), relay.const(4)],
574 if_sample = relay.If(v1, relay.const(1), relay.Tuple([relay.const(2), relay.const(3)]))
575 same = relay.If(v1, relay.const(1), relay.Tuple([relay.const(2), relay.const(3)]))
578 different_cond = relay.If(v2, relay.const(1), relay.Tuple([relay.const(2), relay.const(3)]))
[all …]
H A Dtest_pass_unmatched_cases.py20 from tvm import relay
28 v = relay.Var("v")
39 match = relay.Match(v, [relay.Clause(relay.PatternWildcard(), v)])
44 match = relay.Match(v, [relay.Clause(relay.PatternVar(w), w)])
58 v, [relay.Clause(relay.PatternConstructor(box_ctor, [relay.PatternWildcard()]), v)]
140 relay.Clause(relay.PatternWildcard(), v),
175 relay.Clause(relay.PatternConstructor(p.nil, []), v),
210 relay.Clause(relay.PatternConstructor(p.nil, []), v),
386 relay.Clause(relay.PatternConstructor(p.nil, []), v),
395 clause = relay.Clause(relay.PatternTuple([relay.PatternVar(a), relay.PatternVar(b)]), a + b)
[all …]
H A Dtest_pass_convert_op_layout.py21 from tvm import relay
107 y = relay.Function(relay.analysis.free_vars(y), y)
150 y = relay.Function(relay.analysis.free_vars(y), y)
185 y = relay.Function(relay.analysis.free_vars(y), y)
445 y = relay.add(y, relay.const(1, "float32"))
455 y = relay.add(y, relay.const(1.0, "float32"))
485 beta = relay.var("beta", relay.TensorType((64,), dtype))
577 y = relay.Function(relay.analysis.free_vars(y), y)
623 [relay.const(1, "int32"), relay.const(1, "int32")],
665 [relay.const(1, "int32"), relay.const(1, "int32")],
[all …]
H A Dtest_py_converter.py85 [relay.const(1), relay.const(2), relay.Tuple([relay.const(3), relay.const(4)])]
98 [relay.const(1), relay.const(2), relay.Tuple([relay.const(3), relay.const(4)])]
113 let = relay.Let(v, relay.Tuple([]), relay.Tuple([v, v]))
129 assign = relay.Let(v, relay.RefCreate(relay.Tuple([])), relay.RefRead(v))
139 relay.RefCreate(relay.Tuple([relay.const(1)])),
140 relay.RefWrite(v, relay.Tuple([relay.const(2)])),
150 relay.RefCreate(relay.Tuple([relay.const(1)])),
248 relay.Match(v, [relay.Clause(relay.PatternWildcard(), relay.const(1))]),
261 v, box_ctor(relay.const(1)), relay.Match(v, [relay.Clause(relay.PatternVar(w), w)])
479 relay.Let(x, relay.Tuple([relay.const(1), relay.const(2)]), relay.TupleGetItem(x, 1)),
[all …]
H A Dtest_type_infer.py130 f = relay.Function([x], relay.log(x))
156 sb.ret(f(relay.subtract(n, relay.const(1, ti32)), relay.log(data)))
167 func = relay.Function([x, f], relay.Call(f, [x]), tt)
180 f = relay.Var("f", relay.FuncType([b], b))
216 [f], relay.If(f(), id_func, relay.Function([z], z)), relay.FuncType([b], b), [b]
275 func = relay.Function([x], relay.Call(gv, [x]), tt)
321 v = relay.Var("v", relay.TensorType((), "float32"))
326 relay.PatternConstructor(constructor, [relay.PatternVar(v)]), relay.Tuple([])
329 relay.Clause(relay.PatternWildcard(), relay.Tuple([])),
350 relay.PatternConstructor(constructor, [relay.PatternVar(mv)]), relay.Tuple([])
[all …]
H A Dtest_pass_merge_composite.py20 from tvm import relay, tir
474 x = relay.var("x")
475 w = relay.var("w")
476 b = relay.var("b")
485 x = relay.var("x")
486 w = relay.var("w")
644 branch_1 = relay.multiply(relay.add(x, y), relay.subtract(x, y))
651 branch_2 = relay.multiply(relay.add(x1, y1), relay.subtract(x1, y1))
884 bias = relay.nn.bias_add(conv, relay.var("bias", shape=(256,)))
886 add = relay.op.add(relu, relay.const(1.0))
[all …]
H A Dtest_pass_fuse_ops.py30 y = relay.add(x, relay.const(1, "float32"))
37 y = relay.add(x, relay.const(1, "float32"))
58 x = relay.add(x, relay.const(1, "float32"))
61 y1 = relay.add(relay.const(1, "float32"), y)
73 y = relay.add(x, relay.const(1, "float32"))
81 y1 = relay.add(relay.const(1, "float32"), y)
105 y = relay.Call(f1, [y, relay.var("w1")])
106 z2 = relay.Call(f2, [y, relay.var("w3")])
107 z3 = relay.Call(f3, [y, relay.var("w2"), z2])
198 y = relay.add(x, relay.const(1, "float32"))
[all …]
H A Dtest_ir_parser.py259 relay.Let(X, relay.const(1), relay.Let(Y, relay.const(2), UNIT)),
293 assert_parses_as("(0, 1)", relay.Tuple([relay.const(0), relay.const(1)]))
295 assert_parses_as("(0, 1, 2)", relay.Tuple([relay.const(0), relay.const(1), relay.const(2)]))
352 relay.If(relay.const(True), relay.const(0), relay.const(1)),
381 relay.multiply(relay.const(10), relay.Call(id_func, [relay.const(10)])),
423 relay.Call(multiply, [relay.const(0), relay.const(0)], None, None),
483 relay.Let(relay.Var("_", relay.TensorType((), "float32")), UNIT, UNIT),
544 relay.Let(relay.Var("_", relay.TupleType([])), UNIT, UNIT),
551 relay.Let(relay.Var("_", relay.TupleType([int32])), relay.Tuple([relay.const(0)]), UNIT),
560 relay.Tuple([relay.const(0), relay.const(1)]),
[all …]
H A Dtest_pass_inline.py503 fn1 = relay.Function([], relay.const(1))
505 fn2 = relay.Function([], relay.const(2))
512 mod["main"] = relay.Function([p], relay.Call(relay.If(p, g1, g2), []))
517 fn1 = relay.Function([], relay.const(1))
519 fn2 = relay.Function([], relay.const(2))
522 mod["main"] = relay.Function([p], relay.Call(relay.If(p, fn1, fn2), []))
533 fn1 = relay.Function([], relay.const(1))
536 fn2 = relay.Function([], relay.const(2))
544 mod["main"] = relay.Function([p], relay.Call(relay.If(p, g1, g2), []))
549 fn1 = relay.Function([], relay.const(1))
[all …]
/dports/misc/tvm/incubator-tvm-0.6.1/docs/langref/
H A Drelay_op.rst42 tvm.relay.log
43 tvm.relay.sqrt
45 tvm.relay.exp
47 tvm.relay.add
51 tvm.relay.mod
52 tvm.relay.tanh
106 tvm.relay.abs
139 tvm.relay.all
140 tvm.relay.any
150 tvm.relay.sum
[all …]
/dports/misc/py-tvm/incubator-tvm-0.6.1/docs/langref/
H A Drelay_op.rst42 tvm.relay.log
43 tvm.relay.sqrt
45 tvm.relay.exp
47 tvm.relay.add
51 tvm.relay.mod
52 tvm.relay.tanh
106 tvm.relay.abs
139 tvm.relay.all
140 tvm.relay.any
150 tvm.relay.sum
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/docs/langref/
H A Drelay_op.rst42 tvm.relay.log
43 tvm.relay.sqrt
45 tvm.relay.exp
47 tvm.relay.add
51 tvm.relay.mod
52 tvm.relay.tanh
116 tvm.relay.abs
154 tvm.relay.all
155 tvm.relay.any
166 tvm.relay.sum
[all …]

12345678910>>...168