graph

Documentation

Index

• Variables
• func Asset(name string) ([]byte, error)
• func AssetDir(name string) ([]string, error)
• func AssetInfo(name string) (os.FileInfo, error)
• func AssetNames() []string
• func MustAsset(name string) []byte
• func RestoreAsset(dir, name string) error
• func RestoreAssets(dir, name string) error
• type DataType
  • func GetDataType(s string) DataType
  • func (d DataType) ByteCount() int
  • func (d DataType) String() string
• type Graph
  • func New(path string) (*Graph, error)
  • func (g *Graph) LookUpNodeOperatorName(name string) (string, error)
  • func (g *Graph) MarshalJSON() ([]byte, error)
• type TensorInfo

Variables

var Categories = map[string]string{
	"Const":                            "Constant",
	"Conv2D":                           "Layer",
	"BiasAdd":                          "Layer",
	"DepthwiseConv2dNative":            "Layer",
	"Relu":                             "Activation",
	"Relu6":                            "Activation",
	"Softmax":                          "Activation",
	"Sigmoid":                          "Activation",
	"LRN":                              "Normalization",
	"MaxPool":                          "Pool",
	"MaxPoolV2":                        "Pool",
	"AvgPool":                          "Pool",
	"Reshape":                          "Shape",
	"Squeeze":                          "Shape",
	"ConcatV2":                         "Tensor",
	"Split":                            "Tensor",
	"Dequantize":                       "Tensor",
	"Identity":                         "Control",
	"Variable":                         "Control",
	"VariableV2":                       "Control",
	"Assign":                           "Control",
	"BatchNormWithGlobalNormalization": "Normalization",
	"FusedBatchNorm":                   "Normalization",
}

var OpInfo = []opInfo{}

Functions

func Asset

func Asset(name string) ([]byte, error)

Asset loads and returns the asset for the given name. It returns an error if the asset could not be found or could not be loaded.

func AssetDir

func AssetDir(name string) ([]string, error)

AssetDir returns the file names below a certain directory embedded in the file by go-bindata. For example if you run go-bindata on data/... and data contains the following hierarchy:

data/
  foo.txt
  img/
    a.png
    b.png

then AssetDir("data") would return []string{"foo.txt", "img"} AssetDir("data/img") would return []string{"a.png", "b.png"} AssetDir("foo.txt") and AssetDir("notexist") would return an error AssetDir("") will return []string{"data"}.

func AssetInfo

func AssetInfo(name string) (os.FileInfo, error)

AssetInfo loads and returns the asset info for the given name. It returns an error if the asset could not be found or could not be loaded.

func AssetNames

func AssetNames() []string

AssetNames returns the names of the assets.

func MustAsset

func MustAsset(name string) []byte

MustAsset is like Asset but panics when Asset would return an error. It simplifies safe initialization of global variables.

func RestoreAsset

func RestoreAsset(dir, name string) error

RestoreAsset restores an asset under the given directory

func RestoreAssets

func RestoreAssets(dir, name string) error

RestoreAssets restores an asset under the given directory recursively

Types

type DataType

type DataType tf.DataType

DataType holds the type for a scalar value. E.g., one slot in a tensor.

const (
	Unknown    DataType = DataType(0)
	Float      DataType = DataType(tf.Float)
	Double     DataType = DataType(tf.Double)
	Int32      DataType = DataType(tf.Int32)
	Uint32     DataType = DataType(tf.Uint32)
	Uint8      DataType = DataType(tf.Uint8)
	Int16      DataType = DataType(tf.Int16)
	Int8       DataType = DataType(tf.Int8)
	String     DataType = DataType(tf.String)
	Complex64  DataType = DataType(tf.Complex64)
	Complex    DataType = DataType(tf.Complex)
	Int64      DataType = DataType(tf.Int64)
	Uint64     DataType = DataType(tf.Uint64)
	Bool       DataType = DataType(tf.Bool)
	Qint8      DataType = DataType(tf.Qint8)
	Quint8     DataType = DataType(tf.Quint8)
	Qint32     DataType = DataType(tf.Qint32)
	Bfloat16   DataType = DataType(tf.Bfloat16)
	Qint16     DataType = DataType(tf.Qint16)
	Quint16    DataType = DataType(tf.Quint16)
	Uint16     DataType = DataType(tf.Uint16)
	Complex128 DataType = DataType(tf.Complex128)
	Half       DataType = DataType(tf.Half)
)

func GetDataType

func GetDataType(s string) DataType

func (DataType) ByteCount

func (d DataType) ByteCount() int

func (DataType) String

func (d DataType) String() string

type Graph

type Graph struct {
	*tf.GraphDef
	TensorInfos []TensorInfo `json:"tensor_infos,omitempty"`
	NumBytes    int64        `json:"num_parameters,omitempty"`
}

func New

func New(path string) (*Graph, error)

func (*Graph) LookUpNodeOperatorName

func (g *Graph) LookUpNodeOperatorName(name string) (string, error)

func (*Graph) MarshalJSON

func (g *Graph) MarshalJSON() ([]byte, error)

type TensorInfo

type TensorInfo struct {
	Name              string  `json:"name,omitempty"`
	DataType          string  `json:"data_type,omitempty"`
	DataTypeByteCount int64   `json:"data_type_byte_count,omitempty"`
	ByteCount         int64   `json:"byte_count,omitempty"`
	Dims              []int64 `json:"dims,omitempty"`
	OpName            string  `json:"op_name,omitempty"`
}