Searched refs:transA (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 1388 empty_type_wrapper<float>, const libxsmm_spmdm_handle* handle, char transA, in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread() argument
1392 handle, transA, A, libxsmm_output_csr_a, block_id, tid, nthreads); in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread()
1396 char transA, const bfloat16* A, in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread() argument
1400 handle, transA, reinterpret_cast<const uint16*>(A), libxsmm_output_csr_a, in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread()
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()
1415 empty_type_wrapper<float>, const libxsmm_spmdm_handle* handle, char transA, 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()
|