Searched refs:inputShape (Results 1 – 4 of 4) sorted by relevance
| /test/mlts/benchmark/src/com/android/nn/benchmark/core/ |
| D | TestModels.java | 57 public TestModelEntry(String modelName, float baselineSec, int[] inputShape, in TestModelEntry() argument
64 mInputShape = inputShape; in TestModelEntry()
|
| D | NNTestBase.java | 67 private synchronized native boolean resizeInputTensors(long modelHandle, int[] inputShape); in resizeInputTensors() argument
127 public NNTestBase(String modelName, String modelFile, int[] inputShape, in NNTestBase() argument
142 mInputShape = inputShape; in NNTestBase()
|
| /test/mlts/benchmark/native/ |
| D | multi_process_test.cpp | 191 std::vector<int> inputShape; in main() local
195 inputShape.push_back(std::stoi(currSizeToken)); in main()
213 if (!readInputData(inputDataFileName, inputShape, inputElementSize, in main()
|
| /test/mlts/benchmark/src/com/android/nn/crashtest/core/test/ |
| D | RunModelsInMultipleProcesses.java | 170 List<InferenceInOutSequence> inputOutputAssets, int[] inputShape, int dataByteSize) in writeModelInput() argument
188 for (int dimensionSize : inputShape) { in writeModelInput()
|