Documentation ¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
View Source
var ErrSignatureNotFound = errors.New("no signature with given name found")
ErrSignatureNotFound error returned then given signature was not found from SavedModel
Functions ¶
func GetSignatureDefFromReader ¶
func GetSignatureDefFromReader(tags []string, signature string, r io.Reader) (*protobuf.SignatureDef, error)
GetSignatureDefFromReader returns SignatureDef from a given io.Reader
Types ¶
type Runner ¶
type Runner struct {
// contains filtered or unexported fields
}
Runner transform Session.Run input and output based on given SignatureDef
func NewRunnerWithSignature ¶
func NewRunnerWithSignature(savedModel *tf.SavedModel, signatureDef *protobuf.SignatureDef) (*Runner, error)
NewRunnerWithSignature returns a new Runner for a given SignatureDef and tf.SavedModel
Click to show internal directories.
Click to hide internal directories.