-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathexecution.go
More file actions
75 lines (61 loc) · 2.38 KB
/
execution.go
File metadata and controls
75 lines (61 loc) · 2.38 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
// Code generated by idiomgen. DO NOT EDIT.
package nnapi
import (
"unsafe"
capi "github.com/AndroidGoLab/ndk/capi/neuralnetworks"
)
// Execution wraps the NDK ANeuralNetworksExecution handle.
type Execution struct {
ptr *capi.ANeuralNetworksExecution
}
// cptr returns the underlying C pointer, or nil if h is nil.
// This allows passing optional (nullable) handle parameters to capi functions.
func (h *Execution) cptr() *capi.ANeuralNetworksExecution {
if h == nil {
return nil
}
return h.ptr
}
// NewExecutionFromPointer wraps a raw ANeuralNetworksExecution pointer.
func NewExecutionFromPointer(ptr unsafe.Pointer) *Execution {
return &Execution{ptr: (*capi.ANeuralNetworksExecution)(ptr)}
}
// Pointer returns the underlying pointer as unsafe.Pointer.
func (h *Execution) Pointer() unsafe.Pointer {
return unsafe.Pointer(h.ptr)
}
// UintPtr returns the underlying pointer as a uintptr.
// This is useful for interop with gomobile bind, golang.org/x/mobile,
// gioui.org, and other packages that represent native handles as uintptr.
func (h *Execution) UintPtr() uintptr {
return uintptr(unsafe.Pointer(h.ptr))
}
// NewExecutionFromUintPtr wraps a uintptr as a Execution.
// The caller must ensure ptr points to a valid ANeuralNetworksExecution.
func NewExecutionFromUintPtr(ptr uintptr) *Execution {
return &Execution{ptr: (*capi.ANeuralNetworksExecution)(unsafe.Pointer(ptr))}
}
// Compute calls the underlying NDK function.
func (h *Execution) Compute() error {
return result(capi.ANeuralNetworksExecution_compute(h.ptr))
}
// GetDuration calls the underlying NDK function.
func (h *Execution) GetDuration(durationCode int32, duration *uint64) error {
return result(capi.ANeuralNetworksExecution_getDuration(h.ptr, durationCode, duration))
}
// SetMeasureTiming calls the underlying NDK function.
func (h *Execution) SetMeasureTiming(measure bool) error {
return result(capi.ANeuralNetworksExecution_setMeasureTiming(h.ptr, measure))
}
// SetTimeout calls the underlying NDK function.
func (h *Execution) SetTimeout(duration uint64) error {
return result(capi.ANeuralNetworksExecution_setTimeout(h.ptr, duration))
}
// StartCompute creates a new Event from this Execution.
func (h *Execution) StartCompute() (*Event, error) {
var ptr *capi.ANeuralNetworksEvent
if err := result(capi.ANeuralNetworksExecution_startCompute(h.ptr, &ptr)); err != nil {
return nil, err
}
return &Event{ptr: ptr}, nil
}