Marian NMT
v1.11.5

Contents:

  • Code Organisation
  • Expression graphs
  • Operations in the expression graph
  • Layers
  • Using Marian with factors
  • Library API
  • How to contribute to Marian
  • Writing documentation
Marian NMT
  • »
  • File topk.cu
  • View page source

File topk.cu¶

↰ Parent directory (src/tensors/gpu)

Contents

  • Definition (src/tensors/gpu/topk.cu)

  • Includes

  • Namespaces

  • Functions

  • Defines

  • Variables

Definition (src/tensors/gpu/topk.cu)¶

  • Program Listing for File topk.cu

Includes¶

  • cuda.h

  • tensors/allocator.h (File allocator.h)

  • tensors/gpu/cuda_helpers.h (File cuda_helpers.h)

  • tensors/tensor_operators.h (File tensor_operators.h)

Namespaces¶

  • Namespace marian

  • Namespace marian::gpu

Functions¶

  • Template Function marian::gpu::gMaxElement

  • Template Function marian::gpu::gMaxElementUpdate

  • Function marian::gpu::TopK

Defines¶

  • Define UNROLL_MAXARG_LOOP

Variables¶

  • Variable marian::gpu::BLOCK_SIZE

  • Variable marian::gpu::MAX_BINS


© Copyright 2021, Marian NMT Team.

Built with Sphinx using a theme provided by Read the Docs.