Searched refs:shape (Results 1 – 4 of 4) sorted by relevance
103 bool resizeInputTensors(std::vector<int> shape);
259 bool BenchmarkModel::resizeInputTensors(std::vector<int> shape) { in resizeInputTensors() argument262 mTfliteInterpreter->ResizeInputTensor(input, shape); in resizeInputTensors()
127 std::vector<int> shape(shapePtr, shapePtr + shapeLen); in Java_com_android_nn_benchmark_core_NNTestBase_resizeInputTensors() local128 return model->resizeInputTensors(std::move(shape)); in Java_com_android_nn_benchmark_core_NNTestBase_resizeInputTensors()
248 assert(cpu_tensor.shape == nnapi_tensor.shape)