Paddle

Commit Graph

Author	SHA1	Message	Date
Yu Yang	c532b96741	Remove template parameter for Tensor methods (#4937 ) * Remove template parameter for Tensor methods * Also check the type is correct when data() * Simplize holder_ * Fix accuracy_op * Register Code	7 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.	7 years ago

Author

SHA1

Message

Date

Yu Yang

c532b96741

Remove template parameter for Tensor methods (#4937 )

* Remove template parameter for Tensor methods

* Also check the type is correct when data()
* Simplize holder_

* Fix accuracy_op

* Register Code

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 (2ac5d7d0189c7095c22db68a220be1459abb5486)