Skip to content

Commit ce3a2bb

Browse files
committed
chore: use .reserve() for input shapes
1 parent 3168986 commit ce3a2bb

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

  • packages/react-native-executorch/common/rnexecutorch/models

packages/react-native-executorch/common/rnexecutorch/models/BaseModel.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,6 @@ std::vector<int32_t> BaseModel::getInputShape(std::string method_name,
4141
std::to_string(static_cast<int>(method_meta.error())));
4242
}
4343

44-
std::vector<int32_t> input_shape;
4544
auto input_meta = method_meta->input_tensor_meta(index);
4645
if (!input_meta.ok()) {
4746
throw std::runtime_error(
@@ -50,7 +49,10 @@ std::vector<int32_t> BaseModel::getInputShape(std::string method_name,
5049
std::to_string(static_cast<int>(input_meta.error())));
5150
}
5251

53-
for (auto size : input_meta->sizes()) {
52+
std::vector<int32_t> input_shape;
53+
auto sizes = input_meta->sizes();
54+
input_shape.reserve(sizes.size());
55+
for (auto size : sizes) {
5456
input_shape.push_back(size);
5557
}
5658
return input_shape;

0 commit comments

Comments
 (0)