Searched refs:mInputShape (Results 1 – 3 of 3) sorted by relevance
| /test/mlts/benchmark/src/com/android/nn/benchmark/core/ |
| D | TestModels.java | 37 public final int[] mInputShape; field in TestModels.TestModelEntry 67 mInputShape = inputShape; in TestModelEntry() 79 return new NNTestBase(mModelName, mModelFile, mInputShape, mInOutAssets, mInOutDatasets, in createNNTestBase() 117 return new TestModelEntry(mModelName, mBaselineSec, mInputShape, mInOutAssets, in withDisabledEvaluation()
|
| D | NNTestBase.java | 137 private final int[] mInputShape; field in NNTestBase 196 mInputShape = inputShape; in NNTestBase() 273 if (!resizeInputTensors(mModelHandle, mInputShape)) { in setupModel()
|
| /test/mlts/benchmark/src/com/android/nn/crashtest/core/test/ |
| D | RunModelsInMultipleProcesses.java | 138 if (!writeModelInput(targetInputFile, inputOutputAssets, mTestModelEntry.mInputShape, in call() 145 String inputShapeAsString = Arrays.toString(mTestModelEntry.mInputShape); in call()
|