Function apply_on_physical_tensor

imperative/src/impl/ops/vision.cpp:41–60

}

SmallVector<TensorPtr> apply_on_physical_tensor(
        const OpDef& def, const SmallVector<TensorPtr>& inputs,
        SmallVector<LogicalTensorDesc>& output_descs, const bool& validated) {
    auto&& op = def.cast_final_safe<Flip>();
    auto cn = inputs[0]->comp_node();

    DnnOprCaller<megdnn::Flip> dnn_opr(cn, op.param());
    auto&& out_layout = [&]() -> TensorLayout {
        if (validated) {
            return output_descs[0].layout;
        } else {
            return dnn_opr.deduce_layout(inputs[0]->layout());
        }
    }();

    auto out = Tensor::make(out_layout, cn);

    dnn_opr.exec_with_ws(inputs[0], out);
    return {out};
}

SmallVector<VarNode::LayoutConstraintCallback> get_input_layout_constraint(
        const OpDef& def, const SmallVector<TensorPtr>& inputs) {