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.
Click to show internal directories.
Click to hide internal directories.