Function apply_on_var_node

imperative/src/impl/ops/broadcast.cpp:51–83

    return {descs, success};
}
VarNodeArray apply_on_var_node(const OpDef& def, const VarNodeArray& inputs) {
    auto&& op = def.cast_final_safe<MeshGrid>();
    std::vector<size_t> indexs(inputs.size());
    std::iota(indexs.begin(), indexs.end(), 0);
    auto cn = inputs[0]->comp_node();
    auto graph = inputs[0]->owner_graph();
    if (op.indexing == "xy") {
        if (indexs.size() >= 2) {
            std::swap(indexs[0], indexs[1]);
        }
    } else {
        mgb_assert(op.indexing == "ij", "meshgrid only support \"ij\" or \"xy\"");
    }
    VarNodeArray shps;
    for (size_t ind = 0; ind < inputs.size(); ind++) {
        auto&& inp = inputs[indexs[ind]];
        shps.push_back(opr::GetVarShape::make(inp).node());
    }
    VarNode* tshp = opr::Concat::make(shps, 0, cn).node();
    VarNodeArray results;
    auto t_ndim = inputs.size();
    for (size_t ind = 0; ind < inputs.size(); ind++) {
        auto axis = indexs[ind];
        HostTensorND hv = HostTensorND(cn, {t_ndim}, dtype::Int32());
        auto* ptr = hv.ptr<dt_int32>();
        std::fill_n(ptr, t_ndim, 1);
        ptr[axis] = -1;
        auto shp = opr::ImmutableTensor::make(*graph, hv, cn).node();
        auto tmp = opr::Reshape::make(inputs[ind], shp, axis).node();
        results.push_back(opr::Broadcast::make(tmp, tshp).node());
    }
    return results;
}
SmallVector<TensorPtr> apply_on_physical_tensor(
        const OpDef& def, const SmallVector<TensorPtr>& inputs,
        SmallVector<LogicalTensorDesc>& output_descs, const bool& validated) {