Searched defs:CSRSparseMatrixAdd (Results 1 – 2 of 2) sorted by relevance
275 struct CSRSparseMatrixAdd<GPUDevice, T> struct277 explicit CSRSparseMatrixAdd(OpKernelContext* ctx, const T alpha, const T beta) in CSRSparseMatrixAdd() argument284 Status Initialize() { in Initialize()293 Status GetWorkspaceSize(const ConstCSRComponent<T>& a, in GetWorkspaceSize()319 Status GetOutputStructure(const ConstCSRComponent<T>& a, in GetOutputStructure()349 Status Compute(const ConstCSRComponent<T>& a, const ConstCSRComponent<T>& b, in Compute()375 OpKernelContext* ctx_;376 GpuSparse cuda_sparse_;377 GpuSparseMatrixDescriptor descrA_;378 GpuSparseMatrixDescriptor descrB_;[all …]
187 struct CSRSparseMatrixAdd : public CSRStructureModifyingFunctor<Device, T> { struct