Searched refs:transB (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 1406 char transA, char transB, const bfloat16* alpha, in wrapper_libxsmm_spmdm_compute_generic_thread() argument
1410 handle, transA, transB, reinterpret_cast<const uint16*>(alpha), A_sparse, in wrapper_libxsmm_spmdm_compute_generic_thread()
1416 char transB, const float* alpha, libxsmm_CSR_sparseslice* A_sparse, in wrapper_libxsmm_spmdm_compute_generic_thread() argument
1419 return libxsmm_spmdm_compute_fp32_thread(handle, transA, transB, alpha, in wrapper_libxsmm_spmdm_compute_generic_thread()
|