Home
last modified time | relevance | path

Searched defs:reduce_axis (Results 1 – 25 of 28) sorted by relevance

12

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/te/
H A Dtensor.cc37 IterVar reduce_axis(Range dom, std::string name) { return IterVar(dom, Var(name), kCommReduce); } in reduce_axis() function
124 Array<Region> regions, Array<IterVar> reduce_axis, in TensorIntrinCall()
137 Array<IterVar> reduce_axis, Array<PrimExpr> scalar_inputs) { in __anoncce740ae0502()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/te/
H A Dtensor_intrin.h114 Array<IterVar> reduce_axis; variable
H A Doperation.h208 Array<IterVar> reduce_axis; variable
/dports/misc/tvm/incubator-tvm-0.6.1/include/tvm/
H A Dtensor_intrin.h150 Array<IterVar> reduce_axis; variable
H A Doperation.h216 Array<IterVar> reduce_axis; variable
/dports/misc/py-tvm/incubator-tvm-0.6.1/include/tvm/
H A Dtensor_intrin.h150 Array<IterVar> reduce_axis; variable
H A Doperation.h216 Array<IterVar> reduce_axis; variable
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/te/operation/
H A Dtensor_compute_op.cc57 Array<IterVar> reduce_axis, int schedulable_ndim, in TensorComputeOp()
77 Array<PrimExpr> scalar_inputs) { in __anon672b32e80202()
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/lang/
H A Dtensor.cc116 Array<IterVar> reduce_axis, in make()
H A Dexpr.cc146 IterVar reduce_axis(Range dom, std::string name) { in reduce_axis() function
/dports/misc/tvm/incubator-tvm-0.6.1/src/lang/
H A Dtensor.cc116 Array<IterVar> reduce_axis, in make()
H A Dexpr.cc146 IterVar reduce_axis(Range dom, std::string name) { in reduce_axis() function
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/te/
H A Doperation.py393 def reduce_axis(dom, name="rv"): function
H A Dtensor.py172 def reduce_axis(self): member in BaseComputeOp
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/script/
H A Dintrin.py157 def reduce_axis(begin, end): function
/dports/misc/tvm/incubator-tvm-0.6.1/src/op/
H A Dtensor_compute_op.cc56 Array<IterVar> reduce_axis, in make()
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/op/
H A Dtensor_compute_op.cc56 Array<IterVar> reduce_axis, in make()
/dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/
H A Dtensor.py173 def reduce_axis(self): member in BaseComputeOp
H A Dapi.py747 def reduce_axis(dom, name="rv"): function
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/
H A Dtensor.py173 def reduce_axis(self): member in BaseComputeOp
H A Dapi.py747 def reduce_axis(dom, name="rv"): function
/dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/autotvm/task/
H A Dspace.py633 reduce_axis = axis variable in ConfigSpace
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/autotvm/task/
H A Dspace.py689 reduce_axis = axis variable in ConfigSpace
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/autotvm/task/
H A Dspace.py633 reduce_axis = axis variable in ConfigSpace
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/te/schedule/
H A Dschedule_postproc_rewrite_for_tensor_core.cc285 auto reduce_axis = compute->reduce_axis; in MatrixIdentify() local

12