@Override
    public TensorType type() {
        return model.requireReferencedFunction(function).returnType().get();
    }
The model this is part of
model-evaluation/src/main/java/ai/vespa/models/evaluation/LazyValue.java