Searched refs:inputShape (Results 1 – 2 of 2) sorted by relevance
54 public TestModelEntry(String modelName, float baselineSec, int[] inputShape, in TestModelEntry() argument61 mInputShape = inputShape; in TestModelEntry()
51 private synchronized native boolean resizeInputTensors(long modelHandle, int[] inputShape); in resizeInputTensors() argument92 public NNTestBase(String modelName, String modelFile, int[] inputShape, in NNTestBase() argument107 mInputShape = inputShape; in NNTestBase()