Function elemwise_rule

imperative/src/impl/transformations/format.cpp:462–481

}

ValueRefList elemwise_rule(
        const Elemwise& op, Span<ValueRef>& inputs, const bool& auto_convert,
        const FormatTransformation& t) {
    FT format = get_inputs_format(inputs, t);
    if (format == FT::NHWC && auto_convert) {
        ValueRefList unified_inputs(inputs.size());
        for (size_t i = 0; i < inputs.size(); ++i) {
            auto&& inp = inputs[i].cast(t.value_type());
            if (inp.format() != FT::NHWC && inp.value().is_scalar()) {
                unified_inputs[i] = t.value_type().make(inp.value(), FT::NHWC);
            } else {
                unified_inputs[i] = inputs[i];
            }
        }
        unified_inputs = unify_inputs_format(unified_inputs, FT::NHWC, op.scope(), t);
        return t.wrap_outputs(
                imperative::apply(op, t.unwrap_inputs(unified_inputs)), format);
    }
    return t.wrap_outputs(imperative::apply(op, t.unwrap_inputs(inputs)), format);
}

ValueRefList concat_rule(
        const Concat& op, Span<ValueRef>& inputs, const bool& auto_convert,