Documentation
¶
Index ¶
- Constants
- func InitOperator(op Operator, graphDef []byte, options *tf.SessionOptions) error
- func InitUsingB64Graph(op Operator, b64SerializedGraph string, options *tf.SessionOptions) error
- func InitUsingGraphFile(op Operator, graphFile string, options *tf.SessionOptions) error
- func InitUsingGraphReader(op Operator, graphReader io.Reader, options *tf.SessionOptions) error
- type Op
- type Operator
Constants ¶
View Source
const (
Version = "version"
)
Variables ¶
This section is empty.
Functions ¶
func InitOperator ¶
func InitOperator(op Operator, graphDef []byte, options *tf.SessionOptions) error
InitOperator is initializes operator using graphDef passes as a byte buffer.
func InitUsingB64Graph ¶
func InitUsingB64Graph(op Operator, b64SerializedGraph string, options *tf.SessionOptions) error
InitUsingB64Graph initializes an operator with a graph and a session.
func InitUsingGraphFile ¶
func InitUsingGraphFile(op Operator, graphFile string, options *tf.SessionOptions) error
InitUsingGraphFile reads graph from a file and initializes the operator.
func InitUsingGraphReader ¶
InitUsingGraphReader reads graph using an io reader and initializes the operator.
Types ¶
type Op ¶
type Op struct {
// contains filtered or unexported fields
}
Op is the operator for TF interface.
func (*Op) GetGraph ¶
func (op *Op) GetGraph() *tensorflow.Graph
func (*Op) GetSession ¶
func (op *Op) GetSession() *tensorflow.Session
func (*Op) SetGraph ¶
func (op *Op) SetGraph(graph *tensorflow.Graph)
func (*Op) SetSession ¶
func (op *Op) SetSession(sess *tensorflow.Session)
type Operator ¶
type Operator interface {
// Version returns version info.
Version() (string, error)
// GetGraph retrieves graph.
GetGraph() *tensorflow.Graph
// SetGraph assigns graph to the operator.
SetGraph(*tensorflow.Graph)
// GetSession gets the session.
GetSession() *tensorflow.Session
// SetSession assigns session to the operator.
SetSession(*tensorflow.Session)
}
Operator defines a generic operator to interface with TF.
Source Files
Click to show internal directories.
Click to hide internal directories.