zishuo/onnxruntime_go
1
0
Fork 0
mirror of https://github.com/yalue/onnxruntime_go.git synced 2025-11-02 21:34:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add tensor support

Browse Source 
- There is now a wrapper to create generic Tensor objects in Go, that
   should be appropriately backed by the onnxruntime.

 - Did some minor tests.

 - Still to do: update the example, and put together a nice API that
   uses the tensors.
This commit is contained in:
yalue
2023-02-01 19:47:53 -05:00
parent a8e06273d0
commit ff910beb76
7 changed files with 359 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
onnx_example_application/onnx_example_application.go
View File

@@ -13,7 +13,11 @@ func run() int {
if runtime.GOOS == "windows" {
onnxruntime.SetSharedLibraryPath("../test_data/onnxruntime.dll")
} else {
onnxruntime.SetSharedLibraryPath("../test_data/onnxruntime.so")
if runtime.GOARCH == "arm64" {
onnxruntime.SetSharedLibraryPath("../test_data/onnxruntime_arm64.so")
} else {
onnxruntime.SetSharedLibraryPath("../test_data/onnxruntime.so")
}
}
e := onnxruntime.InitializeEnvironment()
if e != nil {