Options
All
  • Public
  • Public/Protected
  • All
Menu

Class InGPUMemoryShuffledInputProviderBuilder

An in GPU memory ShuffledInputProviderBuilder that shuffles NDArrays on the GPU and keeps them mutually in sync. This is more performant than the CPU version as textures will stay in memory, however this is more GPU memory intensive as it keeps textures resident in GPU memory.

Hierarchy

  • InMemoryShuffledInputProviderBuilder
    • InGPUMemoryShuffledInputProviderBuilder

Implements

  • ShuffledInputProviderBuilder

Index

Constructors

constructor

Properties

Protected epoch

epoch: number = 0

Protected idx

idx: number = 0

Protected inputCounter

inputCounter: number = 0

Protected inputs

inputs: NDArray[][]

All of the inputs, size: [number of inputs][number of examples].

Protected numInputs

numInputs: number

Protected shuffledIndices

shuffledIndices: Uint32Array

Methods

Protected getCurrentExampleIndex

  • getCurrentExampleIndex(): number
  • Returns number

getEpoch

  • getEpoch(): number

getInputProvider

  • getInputProvider(inputId: number): object
  • Parameters

    • inputId: number

    Returns object

getInputProviders

  • Returns input providers which shuffle the inputs and stay in sync.

    Returns InputProvider[]

Protected getNextInput

  • getNextInput(inputId: number): NDArray

Generated using TypeDoc