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
- type Graph
- type TensorInfo
Constants ¶
This section is empty.
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 ¶
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 ¶
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 ¶
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 MustAsset ¶
MustAsset is like Asset but panics when Asset would return an error. It simplifies safe initialization of global variables.
func RestoreAsset ¶
RestoreAsset restores an asset under the given directory
func RestoreAssets ¶
RestoreAssets restores an asset under the given directory recursively
Types ¶
type 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 ¶
type Graph ¶
type Graph struct {
*tf.GraphDef
TensorInfos []TensorInfo `json:"tensor_infos,omitempty"`
NumBytes int64 `json:"num_parameters,omitempty"`
}
Source Files