tf

Documentation

Index

• type Wrapper
  • func NewWrapper(url string, name string, version int) (*Wrapper, error)
  • func (w *Wrapper) Healthy() error
  • func (w *Wrapper) Invoke(nums []int32) ([]int32, error)

Types

type Wrapper

type Wrapper struct {
	// contains filtered or unexported fields
}

Wrapper structure used to call TF grpc service

func NewWrapper

func NewWrapper(url string, name string, version int) (*Wrapper, error)

NewWrapper creates Wrapper

func (*Wrapper) Healthy

func (w *Wrapper) Healthy() error

Healthy return nil or error is TF model is not accesible

func (*Wrapper) Invoke

func (w *Wrapper) Invoke(nums []int32) ([]int32, error)

Invoke is main method