saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-17 01:44:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove temp debug code

Browse source
This commit is contained in:
Scott McKay 2024-12-30 12:01:41 +10:00
parent 019edc9264
commit 6fb01c19a7
2 changed files with 1 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
onnxruntime/core/framework/tensor_type_and_shape.cc
View file

@ -53,14 +53,7 @@ ORT_API_STATUS_IMPL(OrtApis::SetDimensions, OrtTensorTypeAndShapeInfo* info,
_In_ const int64_t* dim_values, size_t dim_count) {
API_IMPL_BEGIN
if (std::any_of(dim_values, dim_values + dim_count, [](int64_t v) { return v < -1; })) {
std::string dim_str = std::string("num_dims=") + std::to_string(dim_count) + " ";
std::for_each(dim_values, dim_values + dim_count, [&dim_str](int64_t v) { dim_str += std::to_string(v) + ", "; });
return OrtApis::CreateStatus(ORT_INVALID_ARGUMENT,
(std::string("dim_values must be -1 (symbolic dimension) or larger. Got:") +
dim_str)
.c_str());
// return OrtApis::CreateStatus(ORT_INVALID_ARGUMENT, "dim_values must be -1 (symbolic dimension) or larger.");
return OrtApis::CreateStatus(ORT_INVALID_ARGUMENT, "dim_values must be -1 (symbolic dimension) or larger.");
}
auto num_dims = std::max(dim_count, info->dim_params.size());

9
onnxruntime/test/shared_lib/test_model_builder_api.cc
View file

@ -474,15 +474,6 @@ TEST(ModelBuilderAPITest, InvalidDimension) {
}
}
TEST(ModelBuilderAPITest, TestSqueezenet) {
static constexpr PATH_TYPE MODEL_URI = TSTR("testdata/squeezenet/model.onnx");
SessionOptions so;
Ort::Session session(*ort_env, MODEL_URI, so);
auto in0 = session.GetInputTypeInfo(0);
}
/*
Tests required