conv2d

Documentation

Overview

Package conv2d defines a benchmark for the Convolutional Layer.

Index

• type Benchmark
  • func NewBenchmark(driver *driver.Driver) *Benchmark
  • func (b *Benchmark) EnableVerification()
  • func (b *Benchmark) Run()
  • func (b *Benchmark) SelectGPU(gpus []int)
  • func (b *Benchmark) SetUnifiedMemory()
  • func (b *Benchmark) Verify()

Types

type Benchmark

type Benchmark struct {
	N, C, H, W int

	KernelChannel, KernelHeight, KernelWidth int
	PadX, PadY                               int
	StrideX, StrideY                         int
	EnableBackward                           bool
	// contains filtered or unexported fields
}

A Benchmark is a benchmark for the Convolutional Layer.

func NewBenchmark

func NewBenchmark(driver *driver.Driver) *Benchmark

NewBenchmark creates a new Conv2D benchmark. It requires the GPU driver as an argument.

func (*Benchmark) EnableVerification

func (b *Benchmark) EnableVerification()

EnableVerification configures the benchmark to verify the result.

func (*Benchmark) Run

func (b *Benchmark) Run()

Run runs the benchmark.

func (*Benchmark) SelectGPU

func (b *Benchmark) SelectGPU(gpus []int)

SelectGPU selects the GPU to run the benchmark on.

func (*Benchmark) SetUnifiedMemory

func (b *Benchmark) SetUnifiedMemory()

SetUnifiedMemory configures the benchmark to use unified memory.

func (*Benchmark) Verify

func (b *Benchmark) Verify()

Verify does nothing for now.