Documentation
¶
Overview ¶
Package tflitecpu runs tflite model files on the host's CPU, as an implementation the ML model service.
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func NewTFLiteCPUModel ¶
func NewTFLiteCPUModel(ctx context.Context, params *TFLiteConfig, name string) (mlmodel.Service, error)
NewTFLiteCPUModel is a constructor that builds a tflite cpu implementation of the MLMS.
Types ¶
type Model ¶
type Model struct {
// contains filtered or unexported fields
}
Model is a struct that implements the TensorflowLite CPU implementation of the MLMS. It includes the configured parameters, model struct, and associated metadata.
type TFLiteConfig ¶
type TFLiteConfig struct { // this should come from the attributes of the tflite_cpu instance of the MLMS ModelPath string `json:"model_path"` NumThreads int `json:"num_threads"` LabelPath *string `json:"label_path"` }
TFLiteConfig contains the parameters specific to a tflite_cpu implementation of the MLMS (machine learning model service).
Click to show internal directories.
Click to hide internal directories.