tf

Documentation

Index

• Constants
• func CallServing(ctx context.Context, servingAddr string, jsonData []byte) ([]byte, error, int)
• func CallTF(ctx context.Context, servingAddr string, model string, version int64, ...) (map[string]interface{}, error, int)
• func NoSuchHostError(err error) bool
• type InputJSON
  • func (t *InputJSON) Tensor() (*tf.TensorProto, error)
• type ModelData

Constants

const (
	MaxMsgLength = 1024 * 1024 * 64 // 64 MB
)

Functions

func CallServing

func CallServing(ctx context.Context, servingAddr string, jsonData []byte) ([]byte, error, int)

func CallTF

func CallTF(ctx context.Context, servingAddr string, model string, version int64,
	signature string, modelData ModelData) (map[string]interface{}, error, int)

func NoSuchHostError

func NoSuchHostError(err error) bool

Types

type InputJSON

type InputJSON struct {
	Dtype tf.DataType `json:"dtype,omitempty"`
	Data  interface{} `json:"data,omitempty"`
}

func (*InputJSON) Tensor

func (t *InputJSON) Tensor() (*tf.TensorProto, error)

type ModelData

type ModelData struct {
	Inputs      map[string]InputJSON `json:"inputs,omitempty"`
	OutFilter   []string             `json:"out_filter,omitempty"`
	OutMimeType string               `json:"out_mime_type,omitempty"`
}