| /dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/relay/op/ |
| H A D | _reduce.py | 28 def _schedule_reduce(_, outs, target): function
34 _reg.register_schedule("argmax", _schedule_reduce)
35 _reg.register_schedule("argmin", _schedule_reduce)
36 _reg.register_schedule("sum", _schedule_reduce)
37 _reg.register_schedule("all", _schedule_reduce)
38 _reg.register_schedule("any", _schedule_reduce)
39 _reg.register_schedule("max", _schedule_reduce)
40 _reg.register_schedule("min", _schedule_reduce)
41 _reg.register_schedule("prod", _schedule_reduce)
42 _reg.register_schedule("mean", _schedule_reduce)
[all …]
|
| H A D | _transform.py | 24 from ._reduce import _schedule_reduce
34 _reg.register_schedule("collapse_sum_like", _schedule_reduce)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/relay/op/ |
| H A D | _reduce.py | 28 def _schedule_reduce(_, outs, target): function
34 _reg.register_schedule("argmax", _schedule_reduce)
35 _reg.register_schedule("argmin", _schedule_reduce)
36 _reg.register_schedule("sum", _schedule_reduce)
37 _reg.register_schedule("all", _schedule_reduce)
38 _reg.register_schedule("any", _schedule_reduce)
39 _reg.register_schedule("max", _schedule_reduce)
40 _reg.register_schedule("min", _schedule_reduce)
41 _reg.register_schedule("prod", _schedule_reduce)
42 _reg.register_schedule("mean", _schedule_reduce)
[all …]
|
| H A D | _transform.py | 24 from ._reduce import _schedule_reduce
34 _reg.register_schedule("collapse_sum_like", _schedule_reduce)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/x86/ |
| H A D | reduction.py | 25 def _schedule_reduce(sch, op, is_idx_reduce=False): function
102 _schedule_reduce(sch, operator, is_idx_reduce=False)
107 _schedule_reduce(sch, operator, is_idx_reduce=True)
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/x86/ |
| H A D | reduction.py | 26 def _schedule_reduce(sch, op, is_idx_reduce=False): function
102 _schedule_reduce(sch, operator, is_idx_reduce=False)
107 _schedule_reduce(sch, operator, is_idx_reduce=True)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/x86/ |
| H A D | reduction.py | 25 def _schedule_reduce(sch, op, is_idx_reduce=False): function
102 _schedule_reduce(sch, operator, is_idx_reduce=False)
107 _schedule_reduce(sch, operator, is_idx_reduce=True)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/nnvm/python/nnvm/top/ |
| H A D | reduction.py | 27 def _schedule_reduce(_, outs, target): function
33 _fschedule_reduce = tvm.convert(_schedule_reduce)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/python/nnvm/top/ |
| H A D | reduction.py | 27 def _schedule_reduce(_, outs, target): function
33 _fschedule_reduce = tvm.convert(_schedule_reduce)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/cuda/ |
| H A D | reduction.py | 25 def _schedule_reduce(op, sch, is_idx_reduce=False): function
133 _schedule_reduce(operator, sch, is_idx_reduce=False)
138 _schedule_reduce(operator, sch, is_idx_reduce=True)
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/cuda/ |
| H A D | reduction.py | 26 def _schedule_reduce(op, sch, is_idx_reduce=False): function
133 _schedule_reduce(operator, sch, is_idx_reduce=False)
138 _schedule_reduce(operator, sch, is_idx_reduce=True)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/cuda/ |
| H A D | reduction.py | 25 def _schedule_reduce(op, sch, is_idx_reduce=False): function
133 _schedule_reduce(operator, sch, is_idx_reduce=False)
138 _schedule_reduce(operator, sch, is_idx_reduce=True)
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/relay/op/ |
| H A D | op.py | 272 return register_schedule(op_name, _schedule_reduce, level)
412 _schedule_reduce = None variable
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/relay/op/strategy/ |
| H A D | generic.py | 88 _op._schedule_reduce = schedule_reduce
|