Optimizing a loop body expressed with tvm.extern

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