Home
last modified time | relevance | path

Searched refs:CSRSparseSparseMatrixMatMul (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dkernels.h194 struct CSRSparseSparseMatrixMatMul struct
196 explicit CSRSparseSparseMatrixMatMul(OpKernelContext* ctx, bool transpose_a,
H A Dsparse_mat_mul_op.cc375 functor::CSRSparseSparseMatrixMatMul<Device, T> csr_gemm( in Compute()
550 struct CSRSparseSparseMatrixMatMul<GPUDevice, T> struct
552 explicit CSRSparseSparseMatrixMatMul(OpKernelContext* ctx, bool transpose_a, in CSRSparseSparseMatrixMatMul() argument
575 ~CSRSparseSparseMatrixMatMul() { in CSRSparseSparseMatrixMatMul() argument