grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
MatMul.forward
grain
functions
binary
MatMul
Undocumented in source. Be warned that the author may not have intended to support it.
auto
forward
(Variable!(T, 2, HostStorage) a, Variable!(T, 2, HostStorage) b)
struct
MatMul
(T)
auto
forward
(
Variable
!(
T
, 2,
HostStorage
)
a
,
Variable
!(
T
, 2,
HostStorage
)
b
)
auto
forward
(Variable!(T, 2, DeviceStorage) a, Variable!(T, 2, DeviceStorage) b)
Meta
Source
See Implementation
grain
functions
binary
MatMul
functions
backward
forward
mixins
__anonymous
variables
alpha
beta
da
db
ha
hb