public static Operand convert()

in wayang-platforms/wayang-tensorflow/src/main/java/org/apache/wayang/tensorflow/model/Convertor.java [38:72]

• 33 lines of code
• 11 McCabe index (conditional complexity)

    public static Operand<?> convert(Ops tf, Op op, Operand<?>... inputs) {
        if (op instanceof ArgMax) {
            return convert(tf, (ArgMax) op, inputs[0]);
        }
        if (op instanceof Cast) {
            return convert(tf, (Cast) op, inputs[0]);
        }
        if (op instanceof CrossEntropyLoss) {
            return convert(tf, (CrossEntropyLoss) op, inputs[0], inputs[1]);
        }
        if (op instanceof Eq) {
            return convert(tf, (Eq) op, inputs[0], inputs[1]);
        }
        if (op instanceof Input) {
            return convert(tf, (Input) op);
        }
        if (op instanceof Linear) {
            return convert(tf, (Linear) op, inputs[0]);
        }
        if (op instanceof Mean) {
            return convert(tf, (Mean) op, inputs[0]);
        }
        if (op instanceof ReLU) {
            return convert(tf, (ReLU) op, inputs[0]);
        }
        if (op instanceof Sigmoid) {
            return convert(tf, (Sigmoid) op, inputs[0]);
        }
        if (op instanceof Softmax) {
            return convert(tf, (Softmax) op, inputs[0]);
        }


        throw new RuntimeException("Unsupported operator: " + op.getClass());
    }