Searched refs:mInputShape (Results 1 – 3 of 3) sorted by relevance
| /test/mlts/benchmark/src/com/android/nn/benchmark/core/ |
| D | TestModels.java | 34 public final int[] mInputShape; field in TestModels.TestModelEntry
64 mInputShape = inputShape; in TestModelEntry()
75 return new NNTestBase(mModelName, mModelFile, mInputShape, mInOutAssets, mInOutDatasets, in createNNTestBase()
102 return new TestModelEntry(mModelName, mBaselineSec, mInputShape, mInOutAssets, in withDisabledEvaluation()
|
| D | NNTestBase.java | 113 private final int[] mInputShape; field in NNTestBase
142 mInputShape = inputShape; in NNTestBase()
191 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()
|