im2col

Documentation

Overview

Package im2col defines a benchmark for the im2col operation.

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

	KernelHeight, KernelWidth int
	PadX, PadY                int
	StrideX, StrideY          int
	DilateX, DilateY          int

	Input tensor.Tensor
	// contains filtered or unexported fields
}

A Benchmark is a benchmark for the im2col operation.

func NewBenchmark

func NewBenchmark(driver *driver.Driver) *Benchmark

NewBenchmark creates a new Im2Col 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.