mirror of
https://github.com/swdee/go-rknnlite.git
synced 2025-12-24 10:30:56 +08:00
add function to return runtime inputtypefloat value
This commit is contained in:
@@ -241,6 +241,11 @@ func (r *Runtime) SetInputTypeFloat32(val bool) {
|
||||
r.inputTypeFloat32 = val
|
||||
}
|
||||
|
||||
// GetInputTypeFloat32 returns the input type if set as Float32 (true) or Int8 (false)
|
||||
func (r *Runtime) GetInputTypeFloat32() bool {
|
||||
return r.inputTypeFloat32
|
||||
}
|
||||
|
||||
// SDKVersion represents the C.rknn_sdk_version struct
|
||||
type SDKVersion struct {
|
||||
DriverVersion string
|
||||
|
||||
Reference in New Issue
Block a user