Fork me on GitHub

src/arraymancer/tensor/tensor_cuda

Theme:

Index

Search:

Group by:

Imports
Exports

Source Edit

Imports

cuda_global_state, cuda, cublas, init_cuda, init_copy_cuda, display_cuda, operators_blas_l1_cuda, operators_blas_l2l3_cuda, operators_broadcasted_cuda, shapeshifting_cuda

Exports

cuda, cpu, zeros_like, ones_like, clone, $, pretty, -=, /, *, /=, +=, +, -, *, *=, dot, *, .+, +.=, -., ./, .=+, .=/, +.=, /., .=-, +., .-, *.=, -., -., +., .*, -.=, /., *., /.=, .=*, +., -.=, squeeze, broadcast2, broadcast, asContiguous, reshape, broadcast, unsqueeze, transpose

Arraymancer Technical reference

Core tensor API

Neural network API

Linear algebra, stats, ML

IO & Datasets

Autograd

Neuralnet primitives

Other docs

Tutorial

Spellbook (How-To's)

Under the hood