File topk.cu

↰ Parent directory (src/tensors/gpu)

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