Searched refs:shape (Results 1 – 4 of 4) sorted by relevance
91 bool resizeInputTensors(std::vector<int> shape);
99 std::vector<int> shape(shapePtr, shapePtr + shapeLen); in Java_com_android_nn_benchmark_core_NNTestBase_resizeInputTensors() local100 return model->resizeInputTensors(std::move(shape)); in Java_com_android_nn_benchmark_core_NNTestBase_resizeInputTensors()
227 bool BenchmarkModel::resizeInputTensors(std::vector<int> shape) { in resizeInputTensors() argument230 mTfliteInterpreter->ResizeInputTensor(input, shape); in resizeInputTensors()
168 assert(cpu_tensor.shape == nnapi_tensor.shape)