ret[irGraph.inputAt(i)] = Nd4j.ones(dtype.nd4jDataType(),*shape)