layer

package
v0.0.0-...-d437eb7 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Jun 18, 2019 License: NCSA Imports: 38 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func ToPythonString 

func ToPythonString(a ast.Ast) (string, error)

Types

type Base 

type Base struct {
	Name_             string `json:"name,omitempty"`
	OperatorType_     string `json:"operator_type,omitempty"`
	OnnxOperatorType_ string `json:"onnx_operator_type,omitempty"`

	InputNames_   []string       `json:"input_names,omitempty"`
	OutputNames_  []string       `json:"output_names,omitempty"`
	InputShapes_  []dlperf.Shape `json:"input_shapes,omitempty"`
	OutputShapes_ []dlperf.Shape `json:"output_shapes,omitempty"`
	// contains filtered or unexported fields
}

func (Base) BwdBenchmarkAlgorithms 

func (b Base) BwdBenchmarkAlgorithms(...dlperf.BwdBenchmarkArgsOptionFunc) []string

func (Base) BwdBenchmarkArgs 

func (b Base) BwdBenchmarkArgs(opts ...dlperf.BwdBenchmarkArgsOptionFunc) interface{}

func (Base) BwdBenchmarkFilter 

func (b Base) BwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.BwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Base) BwdBenchmarkGeneratorArgNames 

func (b Base) BwdBenchmarkGeneratorArgNames() []string

func (Base) BwdBenchmarkName 

func (b Base) BwdBenchmarkName(opts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (Base) DataTypes 

func (c Base) DataTypes() []dlperf.DataType

func (Base) FwdBenchmarkAlgorithms 

func (b Base) FwdBenchmarkAlgorithms(...dlperf.FwdBenchmarkArgsOptionFunc) []string

func (Base) FwdBenchmarkArgs 

func (b Base) FwdBenchmarkArgs(...dlperf.FwdBenchmarkArgsOptionFunc) interface{}

func (Base) FwdBenchmarkFilter 

func (b Base) FwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.FwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Base) FwdBenchmarkGeneratorArgNames 

func (b Base) FwdBenchmarkGeneratorArgNames() []string

func (Base) FwdBenchmarkName 

func (b Base) FwdBenchmarkName(...dlperf.FwdBenchmarkArgsOptionFunc) string

func (Base) InputNames 

func (b Base) InputNames() []string

func (Base) InputShapes 

func (b Base) InputShapes() []dlperf.Shape

func (Base) Inputs 

func (b Base) Inputs() dlperf.Layers

func (Base) MarshalEasyJSON 

func (v Base) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Base) MarshalJSON 

func (v Base) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (*Base) Name 

func (b *Base) Name() string

func (Base) Node 

func (b Base) Node() *onnx.NodeProto

func (Base) OnnxOperatorType 

func (b Base) OnnxOperatorType() string

func (Base) OperatorType 

func (b Base) OperatorType() string

func (Base) OutputNames 

func (b Base) OutputNames() []string

func (Base) OutputShapes 

func (b Base) OutputShapes() []dlperf.Shape

func (Base) Outputs 

func (b Base) Outputs() dlperf.Layers

func (*Base) SetInputNames 

func (b *Base) SetInputNames(names []string)

func (*Base) SetInputShapes 

func (b *Base) SetInputShapes(in []dlperf.Shape)

func (*Base) SetInputs 

func (b *Base) SetInputs(in dlperf.Layers)

func (*Base) SetName 

func (b *Base) SetName(s string)

func (*Base) SetNode 

func (b *Base) SetNode(node *onnx.NodeProto)

func (*Base) SetOnnxOperatorType 

func (b *Base) SetOnnxOperatorType(op string)

func (*Base) SetOperatorType 

func (b *Base) SetOperatorType(s string)

func (*Base) SetOutputNames 

func (b *Base) SetOutputNames(names []string)

func (*Base) SetOutputShapes 

func (b *Base) SetOutputShapes(out []dlperf.Shape)

func (*Base) SetOutputs 

func (b *Base) SetOutputs(out dlperf.Layers)

func (*Base) SetWeightTensors 

func (b *Base) SetWeightTensors(tensors []*onnx.TensorProto)

func (*Base) ShortName 

func (b *Base) ShortName() string

func (*Base) UnmarshalEasyJSON 

func (v *Base) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Base) UnmarshalJSON 

func (v *Base) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

func (Base) WeightTensors 

func (b Base) WeightTensors() []*onnx.TensorProto

type BaseBenchmarkArgs 

type BaseBenchmarkArgs struct {
	ArgNames          []string          `args:"-" json:"arg_names,omitempty"`
	UniqueBenchmarkID uint64            `args:"-" json:"unique_benchmark_id,omitempty"`
	BenchmarkName     string            `args:"-" hash:"name" json:"benchmark_name,omitempty"`
	Algorithms        []string          `args:"-" json:"algorithms,omitempty"`
	DataTypes         []dlperf.DataType `args:"-" json:"data_types,omitempty"`
	IsTraining        bool              `args:"-" json:"is_training,omitempty"`
	BatchSize         int64             `args:"-" json:"batch_size,omitempty"`
}

func (BaseBenchmarkArgs) MarshalEasyJSON 

func (v BaseBenchmarkArgs) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (BaseBenchmarkArgs) MarshalJSON 

func (v BaseBenchmarkArgs) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (*BaseBenchmarkArgs) UnmarshalEasyJSON 

func (v *BaseBenchmarkArgs) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*BaseBenchmarkArgs) UnmarshalJSON 

func (v *BaseBenchmarkArgs) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type BaseBenchmarkInputArgs 

type BaseBenchmarkInputArgs struct {
	Input0    int64 `args:"input[0]" hash:"input[0]" json:"input_0,omitempty"`
	Input1    int64 `args:"input[1]" hash:"input[1]" json:"input_1,omitempty"`
	Input2    int64 `args:"input[2]" hash:"input[2]" json:"input_2,omitempty"`
	Input3    int64 `args:"input[3]" hash:"input[3]" json:"input_3,omitempty"`
	Input4    int64 `args:"input[4]" hash:"input[4]" json:"input_4,omitempty"`
	Input5    int64 `args:"input[5]" hash:"input[5]" json:"input_5,omitempty"`
	Input6    int64 `args:"input[6]" hash:"input[6]" json:"input_6,omitempty"`
	Input7    int64 `args:"input[7]" hash:"input[7]" json:"input_7,omitempty"`
	BatchSize int64 `args:"batch_size" hash:"batch_size" json:"batch_size,omitempty"`
}

func (BaseBenchmarkInputArgs) MarshalEasyJSON 

func (v BaseBenchmarkInputArgs) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (BaseBenchmarkInputArgs) MarshalJSON 

func (v BaseBenchmarkInputArgs) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (*BaseBenchmarkInputArgs) UnmarshalEasyJSON 

func (v *BaseBenchmarkInputArgs) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*BaseBenchmarkInputArgs) UnmarshalJSON 

func (v *BaseBenchmarkInputArgs) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type BatchNorm 

type BatchNorm struct {
	*Base   `json:",inline,flatten,omitempty"`
	Spatial int64 `json:"spatial,omitempty"`
}

https://github.com/onnx/onnx/blob/master/docs/Operators.md#BatchNormalization

func (BatchNorm) BenchmarkAlgorithms 

func (c BatchNorm) BenchmarkAlgorithms() []string

func (BatchNorm) BwdBenchmarkAlgorithms 

func (c BatchNorm) BwdBenchmarkAlgorithms(opts ...dlperf.BwdBenchmarkArgsOptionFunc) []string

func (BatchNorm) BwdBenchmarkArgs 

func (c BatchNorm) BwdBenchmarkArgs(iopts ...dlperf.BwdBenchmarkArgsOptionFunc) interface{}

func (BatchNorm) BwdBenchmarkFilter 

func (c BatchNorm) BwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.BwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (BatchNorm) BwdBenchmarkGenerator 

func (c BatchNorm) BwdBenchmarkGenerator(opts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (BatchNorm) BwdBenchmarkGeneratorArgNames 

func (c BatchNorm) BwdBenchmarkGeneratorArgNames() []string

func (BatchNorm) BwdBenchmarkName 

func (c BatchNorm) BwdBenchmarkName(opts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (BatchNorm) BwdCUDNNName 

func (c BatchNorm) BwdCUDNNName() string

func (BatchNorm) BwdTiming 

func (c BatchNorm) BwdTiming(system string) string

func (BatchNorm) Description 

func (BatchNorm) Description() string

func (BatchNorm) FwdBenchmarkAlgorithms 

func (c BatchNorm) FwdBenchmarkAlgorithms(opts ...dlperf.FwdBenchmarkArgsOptionFunc) []string

func (BatchNorm) FwdBenchmarkArgs 

func (c BatchNorm) FwdBenchmarkArgs(iopts ...dlperf.FwdBenchmarkArgsOptionFunc) interface{}

func (BatchNorm) FwdBenchmarkFilter 

func (c BatchNorm) FwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.FwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (BatchNorm) FwdBenchmarkGenerator 

func (c BatchNorm) FwdBenchmarkGenerator(opts ...dlperf.FwdBenchmarkArgsOptionFunc) string

func (BatchNorm) FwdBenchmarkGeneratorArgNames 

func (c BatchNorm) FwdBenchmarkGeneratorArgNames() []string

func (BatchNorm) FwdBenchmarkName 

func (c BatchNorm) FwdBenchmarkName(iopts ...dlperf.FwdBenchmarkArgsOptionFunc) string

func (BatchNorm) FwdCUDNNName 

func (c BatchNorm) FwdCUDNNName() string

func (BatchNorm) FwdTiming 

func (c BatchNorm) FwdTiming(system string) string

func (*BatchNorm) InferShape 

func (c *BatchNorm) InferShape(inputLayers dlperf.Layers)

func (BatchNorm) Information 

func (c BatchNorm) Information() dlperf.LayerInformation

func (BatchNorm) MarshalEasyJSON 

func (v BatchNorm) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (BatchNorm) MarshalJSON 

func (v BatchNorm) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (BatchNorm) OperatorType 

func (BatchNorm) OperatorType() string

func (BatchNorm) Shape 

func (c BatchNorm) Shape() dlperf.ShapeInformation

func (BatchNorm) ShortName 

func (BatchNorm) ShortName() string

func (*BatchNorm) UnmarshalEasyJSON 

func (v *BatchNorm) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*BatchNorm) UnmarshalJSON 

func (v *BatchNorm) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Cast 

type Cast struct {
	*Base `json:",inline,flatten,omitempty"`
}

func (Cast) Description 

func (Cast) Description() string

func (*Cast) InferShape 

func (c *Cast) InferShape(inputLayers dlperf.Layers)

func (Cast) Information 

func (c Cast) Information() dlperf.LayerInformation

func (Cast) MarshalEasyJSON 

func (v Cast) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Cast) MarshalJSON 

func (v Cast) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Cast) OperatorType 

func (Cast) OperatorType() string

func (Cast) ShortName 

func (Cast) ShortName() string

func (*Cast) UnmarshalEasyJSON 

func (v *Cast) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Cast) UnmarshalJSON 

func (v *Cast) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Clip 

type Clip struct {
	*Base `json:",inline,flatten,omitempty"`
	Min   float32 `json:"min,omitempty"`
	Max   float32 `json:"max,omitempty"`
}

func (Clip) Description 

func (Clip) Description() string

func (*Clip) InferShape 

func (c *Clip) InferShape(inputLayers dlperf.Layers)

func (Clip) Information 

func (c Clip) Information() dlperf.LayerInformation

func (Clip) MarshalEasyJSON 

func (v Clip) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Clip) MarshalJSON 

func (v Clip) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Clip) OperatorType 

func (Clip) OperatorType() string

func (Clip) ShortName 

func (Clip) ShortName() string

func (*Clip) UnmarshalEasyJSON 

func (v *Clip) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Clip) UnmarshalJSON 

func (v *Clip) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Concat 

type Concat struct {
	*Base `json:",inline,flatten,omitempty"`
	Axis  int64 `json:"axis,omitempty"`
}

func (Concat) Description 

func (Concat) Description() string

func (*Concat) InferShape 

func (c *Concat) InferShape(inputLayers dlperf.Layers)

func (Concat) Information 

func (c Concat) Information() dlperf.LayerInformation

func (Concat) MarshalEasyJSON 

func (v Concat) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Concat) MarshalJSON 

func (v Concat) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Concat) OperatorType 

func (Concat) OperatorType() string

func (Concat) ShortName 

func (Concat) ShortName() string

func (*Concat) UnmarshalEasyJSON 

func (v *Concat) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Concat) UnmarshalJSON 

func (v *Concat) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Constant 

type Constant struct {
	*Base `json:",inline,flatten,omitempty"`
}

func (Constant) Description 

func (Constant) Description() string

func (*Constant) InferShape 

func (c *Constant) InferShape(inputLayers dlperf.Layers)

func (Constant) Information 

func (c Constant) Information() dlperf.LayerInformation

func (Constant) MarshalEasyJSON 

func (v Constant) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Constant) MarshalJSON 

func (v Constant) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Constant) OperatorType 

func (Constant) OperatorType() string

func (*Constant) UnmarshalEasyJSON 

func (v *Constant) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Constant) UnmarshalJSON 

func (v *Constant) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type ConstantInput 

type ConstantInput struct {
	*Base `json:",inline,flatten,omitempty"`
}

func (ConstantInput) Description 

func (ConstantInput) Description() string

func (ConstantInput) FwdBenchmarkGenerator 

func (c ConstantInput) FwdBenchmarkGenerator() string

func (*ConstantInput) InferShape 

func (c *ConstantInput) InferShape(inputLayers dlperf.Layers)

func (ConstantInput) Information 

func (c ConstantInput) Information() dlperf.LayerInformation

func (ConstantInput) MarshalEasyJSON 

func (v ConstantInput) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (ConstantInput) MarshalJSON 

func (v ConstantInput) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (ConstantInput) OperatorType 

func (ConstantInput) OperatorType() string

func (*ConstantInput) UnmarshalEasyJSON 

func (v *ConstantInput) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*ConstantInput) UnmarshalJSON 

func (v *ConstantInput) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type ConstantOfShape 

type ConstantOfShape struct {
	*Base `json:",inline,flatten,omitempty"`
}

func (ConstantOfShape) Description 

func (ConstantOfShape) Description() string

func (*ConstantOfShape) InferShape 

func (c *ConstantOfShape) InferShape(inputLayers dlperf.Layers)

func (ConstantOfShape) Information 

func (c ConstantOfShape) Information() dlperf.LayerInformation

func (ConstantOfShape) MarshalEasyJSON 

func (v ConstantOfShape) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (ConstantOfShape) MarshalJSON 

func (v ConstantOfShape) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (ConstantOfShape) OperatorType 

func (ConstantOfShape) OperatorType() string

func (*ConstantOfShape) UnmarshalEasyJSON 

func (v *ConstantOfShape) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*ConstantOfShape) UnmarshalJSON 

func (v *ConstantOfShape) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Conv 

type Conv struct {
	*Base       `json:",inline,flatten,omitempty"`
	AutoPad     string       `json:"auto_pad,omitempty"`
	Dilations   dlperf.Shape `json:"dilation,omitempty"`
	Group       int64        `json:"group,omitempty"`
	KernelShape dlperf.Shape `json:"kernel_shape,omitempty"`
	Pads        dlperf.Shape `json:"pads,omitempty"`
	Strides     dlperf.Shape `json:"strides,omitempty"`
}

func (Conv) BwdBenchmarkAlgorithms 

func (c Conv) BwdBenchmarkAlgorithms(iopts ...dlperf.BwdBenchmarkArgsOptionFunc) []string

func (Conv) BwdBenchmarkArgs 

func (c Conv) BwdBenchmarkArgs(iopts ...dlperf.BwdBenchmarkArgsOptionFunc) interface{}

func (Conv) BwdBenchmarkFilter 

func (c Conv) BwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.BwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Conv) BwdBenchmarkGenerator 

func (c Conv) BwdBenchmarkGenerator(iopts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (Conv) BwdBenchmarkGeneratorArgNames 

func (c Conv) BwdBenchmarkGeneratorArgNames() []string

func (Conv) BwdBenchmarkName 

func (c Conv) BwdBenchmarkName(iopts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (Conv) BwdCUDNNName 

func (c Conv) BwdCUDNNName() string

func (Conv) BwdTiming 

func (c Conv) BwdTiming(system string) string

func (Conv) DataTypes 

func (c Conv) DataTypes() []dlperf.DataType

func (Conv) Description 

func (Conv) Description() string

func (Conv) Flops 

func (c Conv) Flops(algorithm string) dlperf.FlopsInformation

func (Conv) FwdBenchmarkAlgorithms 

func (c Conv) FwdBenchmarkAlgorithms(iopts ...dlperf.FwdBenchmarkArgsOptionFunc) []string

func (Conv) FwdBenchmarkArgs 

func (c Conv) FwdBenchmarkArgs(iopts ...dlperf.FwdBenchmarkArgsOptionFunc) interface{}

func (Conv) FwdBenchmarkFilter 

func (c Conv) FwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.FwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Conv) FwdBenchmarkGenerator 

func (c Conv) FwdBenchmarkGenerator(iopts ...dlperf.FwdBenchmarkArgsOptionFunc) string

func (Conv) FwdBenchmarkGeneratorArgNames 

func (c Conv) FwdBenchmarkGeneratorArgNames() []string

func (Conv) FwdBenchmarkName 

func (c Conv) FwdBenchmarkName(iopts ...dlperf.FwdBenchmarkArgsOptionFunc) string

func (Conv) FwdCUDNNName 

func (c Conv) FwdCUDNNName() string

func (Conv) FwdPythonTensorflow 

func (c Conv) FwdPythonTensorflow() (string, error)

func (Conv) FwdPythonTensorflowAST 

func (c Conv) FwdPythonTensorflowAST() (ast.Ast, error)

func (Conv) FwdTiming 

func (c Conv) FwdTiming(system string) string

func (Conv) HasBias 

func (c Conv) HasBias() bool

func (*Conv) InferShape 

func (c *Conv) InferShape(inputLayers dlperf.Layers)

func (Conv) Information 

func (c Conv) Information() dlperf.LayerInformation

func (Conv) MarshalEasyJSON 

func (v Conv) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Conv) MarshalJSON 

func (v Conv) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Conv) OperatorType 

func (Conv) OperatorType() string

func (Conv) Shape 

func (c Conv) Shape() dlperf.ShapeInformation

func (*Conv) UnmarshalEasyJSON 

func (v *Conv) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Conv) UnmarshalJSON 

func (v *Conv) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Dropout 

type Dropout struct {
	*Base `json:",inline,flatten,omitempty"`
}

func (Dropout) BwdBenchmarkAlgorithms 

func (c Dropout) BwdBenchmarkAlgorithms(...dlperf.BwdBenchmarkArgsOptionFunc) []string

func (Dropout) BwdBenchmarkArgs 

func (c Dropout) BwdBenchmarkArgs(opts ...dlperf.BwdBenchmarkArgsOptionFunc) interface{}

func (Dropout) BwdBenchmarkFilter 

func (c Dropout) BwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.BwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Dropout) BwdBenchmarkGenerator 

func (c Dropout) BwdBenchmarkGenerator(opts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (Dropout) BwdBenchmarkGeneratorArgNames 

func (c Dropout) BwdBenchmarkGeneratorArgNames() []string

func (Dropout) BwdBenchmarkName 

func (c Dropout) BwdBenchmarkName(opts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (Dropout) BwdCUDNNName 

func (c Dropout) BwdCUDNNName() string

func (Dropout) BwdTiming 

func (c Dropout) BwdTiming(system string) string

func (Dropout) Description 

func (Dropout) Description() string

func (Dropout) FwdBenchmarkAlgorithms 

func (c Dropout) FwdBenchmarkAlgorithms(...dlperf.FwdBenchmarkArgsOptionFunc) []string

func (Dropout) FwdBenchmarkArgs 

func (c Dropout) FwdBenchmarkArgs(opts ...dlperf.FwdBenchmarkArgsOptionFunc) interface{}

func (Dropout) FwdBenchmarkFilter 

func (c Dropout) FwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.FwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Dropout) FwdBenchmarkGenerator 

func (c Dropout) FwdBenchmarkGenerator(opts ...dlperf.FwdBenchmarkArgsOptionFunc) string

func (Dropout) FwdBenchmarkGeneratorArgNames 

func (c Dropout) FwdBenchmarkGeneratorArgNames() []string

func (Dropout) FwdBenchmarkName 

func (c Dropout) FwdBenchmarkName(opts ...dlperf.FwdBenchmarkArgsOptionFunc) string

func (Dropout) FwdCUDNNName 

func (c Dropout) FwdCUDNNName() string

func (Dropout) FwdTiming 

func (c Dropout) FwdTiming(system string) string

func (*Dropout) InferShape 

func (c *Dropout) InferShape(inputLayers dlperf.Layers)

func (Dropout) Information 

func (c Dropout) Information() dlperf.LayerInformation

func (Dropout) MarshalEasyJSON 

func (v Dropout) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Dropout) MarshalJSON 

func (v Dropout) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Dropout) OperatorType 

func (Dropout) OperatorType() string

func (Dropout) Shape 

func (c Dropout) Shape() dlperf.ShapeInformation

func (*Dropout) UnmarshalEasyJSON 

func (v *Dropout) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Dropout) UnmarshalJSON 

func (v *Dropout) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type ElementWise 

type ElementWise struct {
	*Base     `json:",inline,flatten,omitempty"`
	Broadcast int64 `json:"broadcast,omitempty"`
	Axis      int64 `json:"axis,omitempty"`
}

func (ElementWise) BwdBenchmarkAlgorithms 

func (c ElementWise) BwdBenchmarkAlgorithms(...dlperf.BwdBenchmarkArgsOptionFunc) []string

func (ElementWise) BwdBenchmarkArgs 

func (c ElementWise) BwdBenchmarkArgs(opts ...dlperf.BwdBenchmarkArgsOptionFunc) interface{}

func (ElementWise) BwdBenchmarkFilter 

func (c ElementWise) BwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.BwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (ElementWise) BwdBenchmarkGenerator 

func (c ElementWise) BwdBenchmarkGenerator(opts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (ElementWise) BwdBenchmarkGeneratorArgNames 

func (c ElementWise) BwdBenchmarkGeneratorArgNames() []string

func (ElementWise) BwdBenchmarkName 

func (c ElementWise) BwdBenchmarkName(opts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (ElementWise) BwdCUDNNName 

func (c ElementWise) BwdCUDNNName() string

func (ElementWise) BwdTiming 

func (c ElementWise) BwdTiming(system string) string

func (ElementWise) Description 

func (ElementWise) Description() string

func (ElementWise) FwdBenchmarkAlgorithms 

func (c ElementWise) FwdBenchmarkAlgorithms(...dlperf.FwdBenchmarkArgsOptionFunc) []string

func (ElementWise) FwdBenchmarkArgs 

func (c ElementWise) FwdBenchmarkArgs(opts ...dlperf.FwdBenchmarkArgsOptionFunc) interface{}

func (ElementWise) FwdBenchmarkFilter 

func (c ElementWise) FwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.FwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (ElementWise) FwdBenchmarkGenerator 

func (c ElementWise) FwdBenchmarkGenerator(opts ...dlperf.FwdBenchmarkArgsOptionFunc) string

func (ElementWise) FwdBenchmarkGeneratorArgNames 

func (c ElementWise) FwdBenchmarkGeneratorArgNames() []string

func (ElementWise) FwdBenchmarkName 

func (c ElementWise) FwdBenchmarkName(opts ...dlperf.FwdBenchmarkArgsOptionFunc) string

func (ElementWise) FwdCUDNNName 

func (c ElementWise) FwdCUDNNName() string

func (ElementWise) FwdTiming 

func (c ElementWise) FwdTiming(system string) string

func (*ElementWise) InferShape 

func (c *ElementWise) InferShape(inputLayers dlperf.Layers)

multidirectionalBroadcastShapeInference

func (ElementWise) Information 

func (c ElementWise) Information() dlperf.LayerInformation

func (ElementWise) MarshalEasyJSON 

func (v ElementWise) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (ElementWise) MarshalJSON 

func (v ElementWise) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (ElementWise) OperatorType 

func (ElementWise) OperatorType() string

func (ElementWise) Shape 

func (c ElementWise) Shape() dlperf.ShapeInformation

func (*ElementWise) UnmarshalEasyJSON 

func (v *ElementWise) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*ElementWise) UnmarshalJSON 

func (v *ElementWise) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Exp 

type Exp struct {
	*Base     `json:",inline,flatten,omitempty"`
	Broadcast int64 `json:"broadcast,omitempty"`
	Axis      int64 `json:"axis,omitempty"`
}

func (Exp) Description 

func (Exp) Description() string

func (*Exp) InferShape 

func (c *Exp) InferShape(inputLayers dlperf.Layers)

multidirectionalBroadcastShapeInference

func (Exp) Information 

func (c Exp) Information() dlperf.LayerInformation

func (Exp) MarshalEasyJSON 

func (v Exp) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Exp) MarshalJSON 

func (v Exp) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Exp) OperatorType 

func (Exp) OperatorType() string

func (*Exp) UnmarshalEasyJSON 

func (v *Exp) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Exp) UnmarshalJSON 

func (v *Exp) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Flatten 

type Flatten struct {
	*Base `json:",inline,flatten,omitempty"`
	Axis  int64 `json:"axis,omitempty"`
}

func (Flatten) Description 

func (Flatten) Description() string

func (*Flatten) InferShape 

func (c *Flatten) InferShape(inputLayers dlperf.Layers)

func (Flatten) Information 

func (c Flatten) Information() dlperf.LayerInformation

func (Flatten) MarshalEasyJSON 

func (v Flatten) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Flatten) MarshalJSON 

func (v Flatten) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Flatten) OperatorType 

func (Flatten) OperatorType() string

func (*Flatten) UnmarshalEasyJSON 

func (v *Flatten) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Flatten) UnmarshalJSON 

func (v *Flatten) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Gather 

type Gather struct {
	*Base `json:",inline,flatten,omitempty"`
	Axis  int64 `json:"axis,omitempty"`
}

func (Gather) Description 

func (Gather) Description() string

func (*Gather) InferShape 

func (c *Gather) InferShape(inputLayers dlperf.Layers)

func (Gather) Information 

func (c Gather) Information() dlperf.LayerInformation

func (Gather) MarshalEasyJSON 

func (v Gather) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Gather) MarshalJSON 

func (v Gather) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Gather) OperatorType 

func (Gather) OperatorType() string

func (*Gather) UnmarshalEasyJSON 

func (v *Gather) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Gather) UnmarshalJSON 

func (v *Gather) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Gemm 

type Gemm struct {
	*Base  `json:",inline,flatten,omitempty"`
	Alpha  float64 `json:"alpha,omitempty"`
	Beta   float64 `json:"beta,omitempty"`
	TransA int64   `json:"transa,omitempty"`
	TransB int64   `json:"transb,omitempty"`
}

func (Gemm) BwdBenchmarkAlgorithms 

func (c Gemm) BwdBenchmarkAlgorithms(...dlperf.BwdBenchmarkArgsOptionFunc) []string

func (Gemm) BwdBenchmarkArgs 

func (c Gemm) BwdBenchmarkArgs(opts ...dlperf.BwdBenchmarkArgsOptionFunc) interface{}

func (Gemm) BwdBenchmarkFilter 

func (c Gemm) BwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.BwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Gemm) BwdBenchmarkGenerator 

func (c Gemm) BwdBenchmarkGenerator(opts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (Gemm) BwdBenchmarkGeneratorArgNames 

func (c Gemm) BwdBenchmarkGeneratorArgNames() []string

func (Gemm) BwdBenchmarkName 

func (c Gemm) BwdBenchmarkName(opts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (Gemm) BwdTiming 

func (c Gemm) BwdTiming(system string) string

func (Gemm) Description 

func (Gemm) Description() string

func (Gemm) FwdBenchmarkAlgorithms 

func (c Gemm) FwdBenchmarkAlgorithms(...dlperf.FwdBenchmarkArgsOptionFunc) []string

func (Gemm) FwdBenchmarkArgs 

func (c Gemm) FwdBenchmarkArgs(opts ...dlperf.FwdBenchmarkArgsOptionFunc) interface{}

func (Gemm) FwdBenchmarkFilter 

func (c Gemm) FwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.FwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Gemm) FwdBenchmarkGenerator 

func (c Gemm) FwdBenchmarkGenerator(opts ...dlperf.FwdBenchmarkArgsOptionFunc) string

func (Gemm) FwdBenchmarkGeneratorArgNames 

func (c Gemm) FwdBenchmarkGeneratorArgNames() []string

func (Gemm) FwdBenchmarkName 

func (c Gemm) FwdBenchmarkName(opts ...dlperf.FwdBenchmarkArgsOptionFunc) string

func (Gemm) FwdTiming 

func (c Gemm) FwdTiming(system string) string

func (*Gemm) InferShape 

func (c *Gemm) InferShape(inputLayers dlperf.Layers)

func (Gemm) Information 

func (c Gemm) Information() dlperf.LayerInformation

func (Gemm) MarshalEasyJSON 

func (v Gemm) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Gemm) MarshalJSON 

func (v Gemm) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Gemm) OperatorType 

func (Gemm) OperatorType() string

func (Gemm) Shape 

func (c Gemm) Shape() dlperf.ShapeInformation

func (*Gemm) UnmarshalEasyJSON 

func (v *Gemm) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Gemm) UnmarshalJSON 

func (v *Gemm) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type GlobalPooling 

type GlobalPooling struct {
	*Base `json:",inline,flatten,omitempty"`
}

func (GlobalPooling) Description 

func (GlobalPooling) Description() string

func (*GlobalPooling) InferShape 

func (c *GlobalPooling) InferShape(inputLayers dlperf.Layers)

func (GlobalPooling) Information 

func (c GlobalPooling) Information() dlperf.LayerInformation

func (GlobalPooling) MarshalEasyJSON 

func (v GlobalPooling) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (GlobalPooling) MarshalJSON 

func (v GlobalPooling) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (GlobalPooling) OperatorType 

func (GlobalPooling) OperatorType() string

func (*GlobalPooling) UnmarshalEasyJSON 

func (v *GlobalPooling) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*GlobalPooling) UnmarshalJSON 

func (v *GlobalPooling) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Identity 

type Identity struct {
	*Base `json:",inline,flatten,omitempty"`
}

func (Identity) Description 

func (Identity) Description() string

func (*Identity) InferShape 

func (c *Identity) InferShape(inputLayers dlperf.Layers)

func (Identity) Information 

func (c Identity) Information() dlperf.LayerInformation

func (Identity) MarshalEasyJSON 

func (v Identity) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Identity) MarshalJSON 

func (v Identity) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Identity) OperatorType 

func (Identity) OperatorType() string

func (*Identity) UnmarshalEasyJSON 

func (v *Identity) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Identity) UnmarshalJSON 

func (v *Identity) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Information 

type Information struct {
	*Base `json:",inline,flatten,omitempty"`
	// contains filtered or unexported fields
}

func (*Information) Flops 

func (info *Information) Flops() dlperf.FlopsInformation

func (*Information) InputNames 

func (info *Information) InputNames() []string

func (Information) MarshalJSON 

func (layer Information) MarshalJSON() ([]byte, error)

func (*Information) Memory 

func (info *Information) Memory() dlperf.MemoryInformation

func (*Information) Name 

func (info *Information) Name() string

func (*Information) OnnxOperatorType 

func (info *Information) OnnxOperatorType() string

func (*Information) OutputNames 

func (info *Information) OutputNames() []string

func (*Information) Shape 

func (info *Information) Shape() dlperf.ShapeInformation

func (*Information) UnmarshalJSON 

func (info *Information) UnmarshalJSON(b []byte) error

func (*Information) Weigths 

func (info *Information) Weigths() []float32

type Int64Slice 

type Int64Slice []int64

Int64Slice attaches the methods of Interface to []int64, sorting in increasing order.

func (Int64Slice) Len 

func (p Int64Slice) Len() int

func (Int64Slice) Less 

func (p Int64Slice) Less(i, j int) bool

func (Int64Slice) Reverse 

func (p Int64Slice) Reverse() Int64Slice

func (Int64Slice) Sort 

func (p Int64Slice) Sort()

Sort is a convenience method.

func (Int64Slice) Swap 

func (p Int64Slice) Swap(i, j int)

type LRN 

type LRN struct {
	*Base `json:",inline,flatten,omitempty"`
	Size  int64 `json:"size,omitempty"`
}

func (LRN) Description 

func (LRN) Description() string

func (*LRN) InferShape 

func (c *LRN) InferShape(inputLayers dlperf.Layers)

func (LRN) Information 

func (c LRN) Information() dlperf.LayerInformation

func (LRN) MarshalEasyJSON 

func (v LRN) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (LRN) MarshalJSON 

func (v LRN) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (LRN) OperatorType 

func (LRN) OperatorType() string

func (*LRN) UnmarshalEasyJSON 

func (v *LRN) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*LRN) UnmarshalJSON 

func (v *LRN) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type NonMaxSuppression 

type NonMaxSuppression struct {
	*Base `json:",inline,flatten,omitempty"`
}

func (NonMaxSuppression) Description 

func (NonMaxSuppression) Description() string

func (*NonMaxSuppression) InferShape 

func (c *NonMaxSuppression) InferShape(inputLayers dlperf.Layers)

func (NonMaxSuppression) Information 

func (c NonMaxSuppression) Information() dlperf.LayerInformation

func (NonMaxSuppression) MarshalEasyJSON 

func (v NonMaxSuppression) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (NonMaxSuppression) MarshalJSON 

func (v NonMaxSuppression) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (NonMaxSuppression) OperatorType 

func (NonMaxSuppression) OperatorType() string

func (*NonMaxSuppression) UnmarshalEasyJSON 

func (v *NonMaxSuppression) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*NonMaxSuppression) UnmarshalJSON 

func (v *NonMaxSuppression) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Pooling 

type Pooling struct {
	*Base       `json:",inline,flatten,omitempty"`
	KernelShape dlperf.Shape `json:"kernel_shape,omitempty"`
	Pads        dlperf.Shape `json:"pads,omitempty"`
	Strides     dlperf.Shape `json:"strides,omitempty"`
}

func (Pooling) BenchmarkAlgorithms 

func (c Pooling) BenchmarkAlgorithms() []string

func (Pooling) BwdBenchmarkAlgorithms 

func (c Pooling) BwdBenchmarkAlgorithms(...dlperf.BwdBenchmarkArgsOptionFunc) []string

func (Pooling) BwdBenchmarkArgs 

func (c Pooling) BwdBenchmarkArgs(opts ...dlperf.BwdBenchmarkArgsOptionFunc) interface{}

func (Pooling) BwdBenchmarkFilter 

func (c Pooling) BwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.BwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Pooling) BwdBenchmarkGenerator 

func (c Pooling) BwdBenchmarkGenerator() string

func (Pooling) BwdBenchmarkGeneratorArgNames 

func (c Pooling) BwdBenchmarkGeneratorArgNames() []string

func (Pooling) BwdBenchmarkName 

func (c Pooling) BwdBenchmarkName(opts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (Pooling) BwdCUDNNName 

func (c Pooling) BwdCUDNNName() string

func (Pooling) BwdTiming 

func (c Pooling) BwdTiming(system string) string

func (Pooling) Description 

func (Pooling) Description() string

func (Pooling) FwdBenchmarkAlgorithms 

func (c Pooling) FwdBenchmarkAlgorithms(...dlperf.FwdBenchmarkArgsOptionFunc) []string

func (Pooling) FwdBenchmarkArgs 

func (c Pooling) FwdBenchmarkArgs(opts ...dlperf.FwdBenchmarkArgsOptionFunc) interface{}

func (Pooling) FwdBenchmarkFilter 

func (c Pooling) FwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.FwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Pooling) FwdBenchmarkGenerator 

func (c Pooling) FwdBenchmarkGenerator() string

func (Pooling) FwdBenchmarkGeneratorArgNames 

func (c Pooling) FwdBenchmarkGeneratorArgNames() []string

func (Pooling) FwdBenchmarkName 

func (c Pooling) FwdBenchmarkName(opts ...dlperf.FwdBenchmarkArgsOptionFunc) string

func (Pooling) FwdCUDNNName 

func (c Pooling) FwdCUDNNName() string

func (Pooling) FwdTiming 

func (c Pooling) FwdTiming(system string) string

func (*Pooling) InferShape 

func (c *Pooling) InferShape(inputLayers dlperf.Layers)

func (Pooling) Information 

func (c Pooling) Information() dlperf.LayerInformation

func (Pooling) MarshalEasyJSON 

func (v Pooling) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Pooling) MarshalJSON 

func (v Pooling) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Pooling) OperatorType 

func (Pooling) OperatorType() string

func (Pooling) Shape 

func (c Pooling) Shape() dlperf.ShapeInformation

func (*Pooling) UnmarshalEasyJSON 

func (v *Pooling) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Pooling) UnmarshalJSON 

func (v *Pooling) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Reduce 

type Reduce struct {
	*Base    `json:",inline,flatten,omitempty"`
	Axes     []int64 `json:"axes,omitempty"`
	KeepDims bool    `json:"keepdims,omitempty"`
}

func (Reduce) Description 

func (Reduce) Description() string

func (*Reduce) InferShape 

func (c *Reduce) InferShape(inputLayers dlperf.Layers)

func (Reduce) Information 

func (c Reduce) Information() dlperf.LayerInformation

func (Reduce) MarshalEasyJSON 

func (v Reduce) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Reduce) MarshalJSON 

func (v Reduce) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Reduce) OperatorType 

func (Reduce) OperatorType() string

func (*Reduce) UnmarshalEasyJSON 

func (v *Reduce) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Reduce) UnmarshalJSON 

func (v *Reduce) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Relu 

type Relu struct {
	*Base `json:",inline,flatten,omitempty"`
}

func (Relu) BenchmarkAlgorithms 

func (c Relu) BenchmarkAlgorithms() []string

func (Relu) BwdBenchmarkAlgorithms 

func (c Relu) BwdBenchmarkAlgorithms(...dlperf.BwdBenchmarkArgsOptionFunc) []string

func (Relu) BwdBenchmarkArgs 

func (c Relu) BwdBenchmarkArgs(opts ...dlperf.BwdBenchmarkArgsOptionFunc) interface{}

func (Relu) BwdBenchmarkFilter 

func (c Relu) BwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.BwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Relu) BwdBenchmarkGenerator 

func (c Relu) BwdBenchmarkGenerator() string

func (Relu) BwdBenchmarkGeneratorArgNames 

func (c Relu) BwdBenchmarkGeneratorArgNames() []string

func (Relu) BwdBenchmarkName 

func (c Relu) BwdBenchmarkName(opts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (Relu) BwdCUDNNName 

func (c Relu) BwdCUDNNName() string

func (Relu) BwdTiming 

func (c Relu) BwdTiming(system string) string

func (Relu) Description 

func (Relu) Description() string

func (Relu) FwdBenchmarkAlgorithms 

func (c Relu) FwdBenchmarkAlgorithms(...dlperf.FwdBenchmarkArgsOptionFunc) []string

func (Relu) FwdBenchmarkArgs 

func (c Relu) FwdBenchmarkArgs(opts ...dlperf.FwdBenchmarkArgsOptionFunc) interface{}

func (Relu) FwdBenchmarkFilter 

func (c Relu) FwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.FwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Relu) FwdBenchmarkGenerator 

func (c Relu) FwdBenchmarkGenerator() string

func (Relu) FwdBenchmarkGeneratorArgNames 

func (c Relu) FwdBenchmarkGeneratorArgNames() []string

func (Relu) FwdBenchmarkName 

func (c Relu) FwdBenchmarkName(opts ...dlperf.FwdBenchmarkArgsOptionFunc) string

func (Relu) FwdCUDNNName 

func (c Relu) FwdCUDNNName() string

func (Relu) FwdTiming 

func (c Relu) FwdTiming(system string) string

func (*Relu) InferShape 

func (c *Relu) InferShape(inputLayers dlperf.Layers)

func (Relu) Information 

func (c Relu) Information() dlperf.LayerInformation

func (Relu) MarshalEasyJSON 

func (v Relu) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Relu) MarshalJSON 

func (v Relu) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Relu) OperatorType 

func (Relu) OperatorType() string

func (Relu) Shape 

func (c Relu) Shape() dlperf.ShapeInformation

func (*Relu) UnmarshalEasyJSON 

func (v *Relu) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Relu) UnmarshalJSON 

func (v *Relu) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Reshape 

type Reshape struct {
	*Base `json:",inline,flatten,omitempty"`
}

func (Reshape) Description 

func (Reshape) Description() string

func (*Reshape) InferShape 

func (c *Reshape) InferShape(inputLayers dlperf.Layers)

func (Reshape) Information 

func (c Reshape) Information() dlperf.LayerInformation

func (Reshape) MarshalEasyJSON 

func (v Reshape) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Reshape) MarshalJSON 

func (v Reshape) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Reshape) OperatorType 

func (Reshape) OperatorType() string

func (*Reshape) UnmarshalEasyJSON 

func (v *Reshape) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Reshape) UnmarshalJSON 

func (v *Reshape) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Scale 

type Scale struct {
	*Base `json:",inline,flatten,omitempty"`
}

func (Scale) Description 

func (Scale) Description() string

func (*Scale) InferShape 

func (c *Scale) InferShape(inputLayers dlperf.Layers)

func (Scale) Information 

func (c Scale) Information() dlperf.LayerInformation

func (Scale) MarshalEasyJSON 

func (v Scale) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Scale) MarshalJSON 

func (v Scale) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Scale) OperatorType 

func (Scale) OperatorType() string

func (*Scale) UnmarshalEasyJSON 

func (v *Scale) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Scale) UnmarshalJSON 

func (v *Scale) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Shape 

type Shape struct {
	*Base `json:",inline,flatten,omitempty"`
}

func (Shape) Description 

func (Shape) Description() string

func (*Shape) InferShape 

func (c *Shape) InferShape(inputLayers dlperf.Layers)

func (Shape) Information 

func (c Shape) Information() dlperf.LayerInformation

func (Shape) MarshalEasyJSON 

func (v Shape) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Shape) MarshalJSON 

func (v Shape) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Shape) OperatorType 

func (Shape) OperatorType() string

func (*Shape) UnmarshalEasyJSON 

func (v *Shape) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Shape) UnmarshalJSON 

func (v *Shape) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Softmax 

type Softmax struct {
	*Base `json:",inline,flatten,omitempty"`
}

func (Softmax) BenchmarkAlgorithms 

func (c Softmax) BenchmarkAlgorithms() []string

func (Softmax) BwdBenchmarkAlgorithms 

func (c Softmax) BwdBenchmarkAlgorithms(...dlperf.BwdBenchmarkArgsOptionFunc) []string

func (Softmax) BwdBenchmarkArgs 

func (c Softmax) BwdBenchmarkArgs(opts ...dlperf.BwdBenchmarkArgsOptionFunc) interface{}

func (Softmax) BwdBenchmarkFilter 

func (c Softmax) BwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.BwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Softmax) BwdBenchmarkGenerator 

func (c Softmax) BwdBenchmarkGenerator() string

func (Softmax) BwdBenchmarkGeneratorArgNames 

func (c Softmax) BwdBenchmarkGeneratorArgNames() []string

func (Softmax) BwdBenchmarkName 

func (c Softmax) BwdBenchmarkName(opts ...dlperf.BwdBenchmarkArgsOptionFunc) string

func (Softmax) BwdCUDNNName 

func (c Softmax) BwdCUDNNName() string

func (Softmax) BwdTiming 

func (c Softmax) BwdTiming(system string) string

func (Softmax) Description 

func (Softmax) Description() string

func (Softmax) FwdBenchmarkAlgorithms 

func (c Softmax) FwdBenchmarkAlgorithms(...dlperf.FwdBenchmarkArgsOptionFunc) []string

func (Softmax) FwdBenchmarkArgs 

func (c Softmax) FwdBenchmarkArgs(opts ...dlperf.FwdBenchmarkArgsOptionFunc) interface{}

func (Softmax) FwdBenchmarkFilter 

func (c Softmax) FwdBenchmarkFilter(datatype, algorithm string, opts ...dlperf.FwdBenchmarkArgsOptionFunc) benchmark.Benchmark

func (Softmax) FwdBenchmarkGenerator 

func (c Softmax) FwdBenchmarkGenerator() string

func (Softmax) FwdBenchmarkGeneratorArgNames 

func (c Softmax) FwdBenchmarkGeneratorArgNames() []string

func (Softmax) FwdBenchmarkName 

func (c Softmax) FwdBenchmarkName(opts ...dlperf.FwdBenchmarkArgsOptionFunc) string

func (Softmax) FwdCUDNNName 

func (c Softmax) FwdCUDNNName() string

func (Softmax) FwdTiming 

func (c Softmax) FwdTiming(system string) string

func (*Softmax) InferShape 

func (c *Softmax) InferShape(inputLayers dlperf.Layers)

func (Softmax) Information 

func (c Softmax) Information() dlperf.LayerInformation

func (Softmax) MarshalEasyJSON 

func (v Softmax) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Softmax) MarshalJSON 

func (v Softmax) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Softmax) OperatorType 

func (Softmax) OperatorType() string

func (Softmax) Shape 

func (c Softmax) Shape() dlperf.ShapeInformation

func (*Softmax) UnmarshalEasyJSON 

func (v *Softmax) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Softmax) UnmarshalJSON 

func (v *Softmax) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Squeeze 

type Squeeze struct {
	*Base `json:",inline,flatten,omitempty"`
	Axes  []int64 `json:"axes,omitempty"`
}

func (Squeeze) Description 

func (Squeeze) Description() string

func (*Squeeze) InferShape 

func (c *Squeeze) InferShape(inputLayers dlperf.Layers)

func (Squeeze) Information 

func (c Squeeze) Information() dlperf.LayerInformation

func (Squeeze) MarshalEasyJSON 

func (v Squeeze) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Squeeze) MarshalJSON 

func (v Squeeze) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Squeeze) OperatorType 

func (Squeeze) OperatorType() string

func (*Squeeze) UnmarshalEasyJSON 

func (v *Squeeze) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Squeeze) UnmarshalJSON 

func (v *Squeeze) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type TopK 

type TopK struct {
	*Base `json:",inline,flatten,omitempty"`
	K     int64 `json:"k,omitempty"`
	Axis  int64 `json:"axis,omitempty"`
}

func (TopK) Description 

func (TopK) Description() string

func (*TopK) InferShape 

func (c *TopK) InferShape(inputLayers dlperf.Layers)

func (TopK) Information 

func (c TopK) Information() dlperf.LayerInformation

func (TopK) MarshalEasyJSON 

func (v TopK) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (TopK) MarshalJSON 

func (v TopK) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (TopK) OperatorType 

func (TopK) OperatorType() string

func (*TopK) UnmarshalEasyJSON 

func (v *TopK) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*TopK) UnmarshalJSON 

func (v *TopK) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Transpose 

type Transpose struct {
	*Base       `json:",inline,flatten,omitempty"`
	Permutation []int64 `json:"perm,omitempty"`
}

func (Transpose) Description 

func (Transpose) Description() string

func (*Transpose) InferShape 

func (c *Transpose) InferShape(inputLayers dlperf.Layers)

func (Transpose) Information 

func (c Transpose) Information() dlperf.LayerInformation

func (Transpose) MarshalEasyJSON 

func (v Transpose) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Transpose) MarshalJSON 

func (v Transpose) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Transpose) OperatorType 

func (Transpose) OperatorType() string

func (*Transpose) UnmarshalEasyJSON 

func (v *Transpose) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Transpose) UnmarshalJSON 

func (v *Transpose) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

type Unsqueeze 

type Unsqueeze struct {
	*Base `json:",inline,flatten,omitempty"`
	Axes  []int64 `json:"axes,omitempty"`
}

func (Unsqueeze) Description 

func (Unsqueeze) Description() string

func (*Unsqueeze) InferShape 

func (c *Unsqueeze) InferShape(inputLayers dlperf.Layers)

func (Unsqueeze) Information 

func (c Unsqueeze) Information() dlperf.LayerInformation

func (Unsqueeze) MarshalEasyJSON 

func (v Unsqueeze) MarshalEasyJSON(w *jwriter.Writer)

MarshalEasyJSON supports easyjson.Marshaler interface

func (Unsqueeze) MarshalJSON 

func (v Unsqueeze) MarshalJSON() ([]byte, error)

MarshalJSON supports json.Marshaler interface

func (Unsqueeze) OperatorType 

func (Unsqueeze) OperatorType() string

func (*Unsqueeze) UnmarshalEasyJSON 

func (v *Unsqueeze) UnmarshalEasyJSON(l *jlexer.Lexer)

UnmarshalEasyJSON supports easyjson.Unmarshaler interface

func (*Unsqueeze) UnmarshalJSON 

func (v *Unsqueeze) UnmarshalJSON(data []byte) error

UnmarshalJSON supports json.Unmarshaler interface

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.