Tensor compute(Array<PrimExpr> shape, FCompute fcompute, std::string name, std::string tag,
Map<String, ObjectRef> attrs) {
- auto op_node = make_object<ComputeOpNode>();
// compute dimension.
size_t ndim = shape.size();
std::vector<IterVar> axis;
Array<Tensor> compute(Array<PrimExpr> shape, FBatchCompute fcompute, std::string name,
std::string tag, Map<String, ObjectRef> attrs) {
- auto op_node = make_object<ComputeOpNode>();
// compute dimension.
size_t ndim = shape.size();
std::vector<IterVar> axis;