Paddle

Commit Graph

Author	SHA1	Message	Date
kexinzhao	cb0118f3e5	Polish Operator Doc (m) (#5375 ) * fix m_ops * fix activation op	8 years ago
Markus Kliegl	164898277c	MatMul operator (#4856 ) * initial matmul operator Similar to np.matmul, but also has transpose_X and transpose_Y flags, and only supports tensors from rank 1 to 3 inclusive. For GPU, uses cublas?gemmStridedBatched. For CPU, uses cblas_?gemm_batch if available via MKL; otherwise a simple serial implementation that loops over the batch dimension is employed for now.	8 years ago

Author

SHA1

Message

Date

kexinzhao

cb0118f3e5

Polish Operator Doc (m) (#5375 )

* fix m_ops

* fix activation op

Markus Kliegl

164898277c

MatMul operator (#4856 )

* initial matmul operator

Similar to np.matmul, but also has transpose_X and transpose_Y flags,
and only supports tensors from rank 1 to 3 inclusive.

For GPU, uses cublas?gemmStridedBatched. For CPU, uses
cblas_?gemm_batch if available via MKL; otherwise a simple serial
implementation that loops over the batch dimension is employed for now.

2 Commits (748fdbbec5a0458a33b660e5cfbb02bf80cd92e5)