tfmodel

Documentation

Index

• func Export(session *tf.Session, graph *tf.Graph, name string) (interface{}, error)
• func MatchOperation(graph *tf.Graph, name string) string
• func RunExport(session *tf.Session, exportOp *tf.Operation) (interface{}, error)
• func Signature(model *tf.SavedModel) (*domain.Signature, error)
• type Evaluator
  • func NewEvaluator(signature *domain.Signature, session *tf.Session) *Evaluator
  • func (e *Evaluator) Close() error
  • func (e *Evaluator) Evaluate(params []interface{}) ([]interface{}, error)

Functions

func Export

func Export(session *tf.Session, graph *tf.Graph, name string) (interface{}, error)

Export run model export, or error

func MatchOperation

func MatchOperation(graph *tf.Graph, name string) string

func RunExport

func RunExport(session *tf.Session, exportOp *tf.Operation) (interface{}, error)

RunExport runs export

func Signature

func Signature(model *tf.SavedModel) (*domain.Signature, error)

Signature returns model signature or error

Types

type Evaluator

type Evaluator struct {
	domain.Signature
	// contains filtered or unexported fields
}

Evaluator represents evaluator

func NewEvaluator

func NewEvaluator(signature *domain.Signature, session *tf.Session) *Evaluator

NewEvaluator creates new evaluator

func (*Evaluator) Close

func (e *Evaluator) Close() error

Close closes evaluator

func (*Evaluator) Evaluate

func (e *Evaluator) Evaluate(params []interface{}) ([]interface{}, error)

Evaluate evaluates model