saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-19 19:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix the tensorflow performance test (#3847)

Browse source
This commit is contained in:
Changming Sun 2020-05-13 11:52:59 -07:00 committed by GitHub
parent 7c774e967a
commit e86214e5c0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
onnxruntime/test/onnx/heap_buffer.h
View file

@ -4,6 +4,7 @@
#pragma once
#include <vector>
#include <memory>
#include <stdlib.h>
namespace onnxruntime {
namespace test {

2
onnxruntime/test/perftest/main.cc
View file

@ -8,12 +8,14 @@
#include "performance_runner.h"
using namespace onnxruntime;
const OrtApi* g_ort = NULL;
#ifdef _WIN32
int real_main(int argc, wchar_t* argv[]) {
#else
int real_main(int argc, char* argv[]) {
#endif
g_ort = OrtGetApiBase()->GetApi(ORT_API_VERSION);
perftest::PerformanceTestConfig test_config;
if (!perftest::CommandLineParser::ParseArguments(test_config, argc, argv)) {
perftest::CommandLineParser::ShowUsage();

13
onnxruntime/test/perftest/tf_test_session.h
View file

@ -7,6 +7,7 @@
#include "test_configuration.h"
#include "tensorflow/c/c_api.h"
#include "test_session.h"
extern const OrtApi* g_ort;
namespace onnxruntime {
namespace perftest {
@ -117,21 +118,21 @@ class TensorflowTestSession : public TestSession {
TF_Status* s = TF_NewStatus();
void* input_buffer = nullptr;
ORT_THROW_ON_ERROR(OrtGetTensorMutableData(const_cast<OrtValue*>(value), &input_buffer));
Ort::ThrowOnError(g_ort->GetTensorMutableData(value, &input_buffer));
assert(input_buffer != nullptr);
OrtTensorTypeAndShapeInfo* shape = nullptr;
ORT_THROW_ON_ERROR(OrtGetTensorTypeAndShape(value, &shape));
Ort::ThrowOnError(g_ort->GetTensorTypeAndShape(value, &shape));
size_t buffer_length = 0;
std::vector<int64_t> dims;
size_t dim_count;
ORT_THROW_ON_ERROR(OrtGetDimensionsCount(shape, &dim_count));
Ort::ThrowOnError(g_ort->GetDimensionsCount(shape, &dim_count));
dims.resize(dim_count);
ORT_THROW_ON_ERROR(OrtGetDimensions(shape, dims.data(), dim_count));
Ort::ThrowOnError(g_ort->GetDimensions(shape, dims.data(), dim_count));
size_t ele_count;
ORT_THROW_ON_ERROR(OrtGetTensorShapeElementCount(shape, &ele_count));
Ort::ThrowOnError(g_ort->GetTensorShapeElementCount(shape, &ele_count));
TF_DataType tf_datatype;
ONNXTensorElementDataType element_type;
ORT_THROW_ON_ERROR(OrtGetTensorElementType(shape, &element_type));
Ort::ThrowOnError(g_ort->GetTensorElementType(shape, &element_type));
switch (element_type) {
case ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT: // maps to c type float
buffer_length = ele_count * sizeof(float);