override fun run(inputs: Map<String, INDArray>): Map<String, INDArray> {
        return graphRunner.run(inputs)
    }