The documentation states the following, and I believe axis
and reduce_axis
are not defined when the tensor is anything other than tvm.tensor.ComputeOp
.
class tvm.tensor.ComputeOp(handle) Bases: tvm.tensor.Operation
Compute operation.
axis Represent axis of IterVar, only defined when it is a ComputeOp
reduce_axis Represent axis of reductions, only defined when it is a ComputeOp