Searched refs:shape (Results 1 – 3 of 3) sorted by relevance
72 bool resizeInputTensors(std::vector<int> shape);
202 bool BenchmarkModel::resizeInputTensors(std::vector<int> shape) { in resizeInputTensors() argument205 mTfliteInterpreter->ResizeInputTensor(input, shape); in resizeInputTensors()
80 std::vector<int> shape(shapePtr, shapePtr + shapeLen); in Java_com_android_nn_benchmark_core_NNTestBase_resizeInputTensors() local81 return model->resizeInputTensors(std::move(shape)); in Java_com_android_nn_benchmark_core_NNTestBase_resizeInputTensors()