generator

Documentation

Index

• Constants
• type Builder
  • func (b *Builder) Build() Generator
  • func (b *Builder) SetMetaGraphTags(metaGraphTags []string)
  • func (b *Builder) SetName(name string)
  • func (b *Builder) SetSigDefKey(sigDefKey string)
  • func (b *Builder) SetVersion(version string)
• type Generator
  • func (g *Generator) GenerateOpenAPI(model *pb.SavedModel) (string, error)

Constants

const (
	DefaultSigDefKey = "serving_default"
	DefaultTag       = "serve"
)

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

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)