Home
last modified time | relevance | path

Searched refs:GpuSparseMatrixDescriptor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dcuda_sparse.h467 class GpuSparseMatrixDescriptor {
469 explicit GpuSparseMatrixDescriptor() : initialized_(false) {} in GpuSparseMatrixDescriptor() function
471 GpuSparseMatrixDescriptor(GpuSparseMatrixDescriptor&& rhs) in GpuSparseMatrixDescriptor() function
476 GpuSparseMatrixDescriptor& operator=(GpuSparseMatrixDescriptor&& rhs) {
485 ~GpuSparseMatrixDescriptor() { Release(); } in ~GpuSparseMatrixDescriptor()
525 TF_DISALLOW_COPY_AND_ASSIGN(GpuSparseMatrixDescriptor);
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dadd_op.cc377 GpuSparseMatrixDescriptor descrA_;
378 GpuSparseMatrixDescriptor descrB_;
379 GpuSparseMatrixDescriptor descrC_;
H A Dsparse_mat_mul_op.cc730 GpuSparseMatrixDescriptor descrA_; in CSRSparseSparseMatrixMatMul()
731 GpuSparseMatrixDescriptor descrB_; in CSRSparseSparseMatrixMatMul()
732 GpuSparseMatrixDescriptor descrC_; in CSRSparseSparseMatrixMatMul()