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

fix json format (#11046)

Browse source 
Co-authored-by: Randy Shuai <rashuai@microsoft.com>
This commit is contained in:
RandySheriffH 2022-03-30 16:15:33 -07:00 committed by GitHub
parent 9616ad483f
commit 9505e8c6c1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
onnxruntime/core/framework/sequential_executor.cc
View file

@ -59,6 +59,7 @@ static void CalculateTotalOutputSizes(OpKernelContextInternal* op_kernel_context
size_t& total_output_sizes, const std::string& node_name, std::string& output_type_shape) {
// Calculate total output sizes for this operation.
std::stringstream ss;
int added_type_shapes = 0;
ss << "[";
total_output_sizes = 0;
ORT_UNUSED_PARAMETER(node_name);
@ -78,8 +79,9 @@ static void CalculateTotalOutputSizes(OpKernelContextInternal* op_kernel_context
#endif
total_output_sizes += tensor_size;
auto shape_str = tensor.Shape().ToString();
ss << "{\"" << DataTypeImpl::ToString(tensor.DataType()) << "\":["
<< shape_str.substr(1, shape_str.size() - 2) << "]" << (i == output_count - 1 ? "}" : "},");
ss << (added_type_shapes++ > 0 ? "," : "")
<< "{\"" << DataTypeImpl::ToString(tensor.DataType()) << "\":["
<< shape_str.substr(1, shape_str.size() - 2) << "]}";
}
}
ss << "]";
@ -93,6 +95,7 @@ static void CalculateTotalInputSizes(const OpKernelContextInternal* op_kernel_co
// Calculate total input sizes for this operation.
std::stringstream ss;
ss << "[";
int added_type_shapes = 0;
input_activation_sizes = 0;
input_parameter_sizes = 0;
ORT_UNUSED_PARAMETER(node_name);
@ -124,8 +127,9 @@ static void CalculateTotalInputSizes(const OpKernelContextInternal* op_kernel_co
input_activation_sizes += tensor_size;
}
auto shape_str = p_tensor->Shape().ToString();
ss << "{\"" << DataTypeImpl::ToString(p_tensor->DataType()) << "\":["
<< shape_str.substr(1, shape_str.size() - 2) << "]" << (i == input_count - 1 ? "}" : "},");
ss << (added_type_shapes++ > 0 ? "," : "")
<< "{\"" << DataTypeImpl::ToString(p_tensor->DataType()) << "\":["
<< shape_str.substr(1, shape_str.size() - 2) << "]}";
}
}
ss << "]";

3
tools/perf_view/ort_perf_view.html
View file

@ -1,7 +1,7 @@
<!DOCTYPE html>
<html>
<head>
<title>My First JavaScript Treemap Chart</title>
<title id="filename">Onnxruntime Perf View</title>
<script type="text/javascript" src="https://cdn.anychart.com/releases/v8/js/anychart-core.min.js"></script>
<script src="https://cdn.anychart.com/releases/8.11.0/js/anychart-ui.min.js"></script>
<script type="text/javascript" src="https://cdn.anychart.com/releases/v8/js/anychart-treemap.min.js"></script>
@ -60,6 +60,7 @@
preloader.visible(true);
const fileList = event.target.files;
document.getElementById('filename').text = fileList[0].name;
console.log(fileList);
const reader = new FileReader();
reader.addEventListener('load', (event) => {