Home
last modified time | relevance | path

Searched defs:CSRSparseMatrixMatMul (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dkernels.h146 struct CSRSparseMatrixMatMul { struct
H A Dmat_mul_op.cc780 class CSRSparseMatrixMatMul<GPUDevice, T> { class
782 explicit CSRSparseMatrixMatMul(const bool transpose_output) in CSRSparseMatrixMatMul() function in tensorflow::functor::CSRSparseMatrixMatMul