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
»
Library API
»
Function marian::gpu::CSRProd
View page source
Function marian::gpu::CSRProd
¶
Defined in
File prod_sparse.cpp
Function Documentation
¶
void
marian::
gpu
::
CSRProd
(
marian::Tensor
C
,
Ptr
<
Allocator
>
allocator
,
const
marian::Tensor &
A_values
,
const
marian::Tensor &
A_indices
,
const
marian::Tensor &
A_offsets
,
const
marian::Tensor &
B
, bool
transA
, bool
swapOperands
, float
beta
= 0
)
¶