Function apply_on_physical_tensor

imperative/src/impl/ops/custom_opdef.cpp:168–200

namespace custom_opdef {  // avoid name conflict

SmallVector<TensorPtr> apply_on_physical_tensor(
        const OpDef& def, const SmallVector<TensorPtr>& inputs,
        SmallVector<LogicalTensorDesc>& output_descs, const bool& validated) {
    SmallVector<TensorPtr> outputs(output_descs.size());

    if (validated == false) {
        auto&& op = static_cast<const CustomOpDef&>(def);
        for (size_t i = 0; i < outputs.size(); ++i) {
            auto [output_descs, success] = op.infer_output_attrs(inputs);
            mgb_assert(success == true, "infer output attributes fall\n");
        }
    }

    for (size_t i = 0; i < outputs.size(); ++i) {
        auto& output = outputs[i];
        output = Tensor::make(output_descs[i].layout, output_descs[i].comp_node);
    }

    std::shared_ptr<SmallVector<DeviceTensorND>> inp_tensornds =
            std::make_shared<SmallVector<DeviceTensorND>>();
    std::shared_ptr<SmallVector<DeviceTensorND>> oup_tensornds =
            std::make_shared<SmallVector<DeviceTensorND>>();
    for (size_t i = 0; i < inputs.size(); ++i) {
        inp_tensornds->emplace_back(inputs[i]->dev_tensor(true));
    }
    for (size_t i = 0; i < outputs.size(); ++i) {
        oup_tensornds->emplace_back(outputs[i]->dev_tensor(true));
    }

    auto&& op = static_cast<const CustomOpDef&>(def);
    op.compute(inp_tensornds, oup_tensornds);
    return outputs;
}

VarNodeArray apply_on_var_node(const OpDef& def, const cg::VarNodeArray& inputs) {
    auto&& op = static_cast<const CustomOpDef&>(def);