Documentation

Index

Constants

View Source
const (
	DefaultSigDefKey = "serving_default"
	DefaultTag       = "serve"
)
View Source
const (
	SpecGenerationError     = "missing info to generate OpenAPI specification\n error: %s"
	UnmarshallableSpecError = "generated OpenAPI specification is corrupted\n error: %s \n specification: %s"
	UnloadableSpecError     = "generated OpenAPI specification (below) is corrupted\n error: %s \n specification: %s"
	InvalidSpecError        = "generated OpenAPI specification (below) is constructed incorrectly\n error: %s \n specification: %s"
)

    Known error messages

    Variables

    This section is empty.

    Functions

    This section is empty.

    Types

    type Builder

    type Builder struct {
    	Generator
    }

    func (*Builder) Build

    func (b *Builder) Build() Generator

    func (*Builder) SetMetaGraphTags

    func (b *Builder) SetMetaGraphTags(metaGraphTags []string)

    func (*Builder) SetName

    func (b *Builder) SetName(name string)

    func (*Builder) SetSigDefKey

    func (b *Builder) SetSigDefKey(sigDefKey string)

    func (*Builder) SetVersion

    func (b *Builder) SetVersion(version string)

    type Generator

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

    func (*Generator) GenerateOpenAPI

    func (g *Generator) GenerateOpenAPI(model *pb.SavedModel) (string, error)