Paddle

Commit Graph

Author	SHA1	Message	Date
xuwei06	6a6e4d8d0f	Fix matmal_op for debug mode The dimension is not set correctly and is not being checked in release mode because eigen_assert is not enabled.	7 years ago
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

xuwei06

6a6e4d8d0f

Fix matmal_op for debug mode

The dimension is not set correctly and is not being checked in release mode because eigen_assert is not enabled.

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.

3 Commits (e0e3a8a5bb2a33bf1953c7cebdedcf2ea5869b51)