Searched refs:CSRSparseMatrixAdd (Results 1 – 2 of 2) sorted by relevance
187 struct CSRSparseMatrixAdd : public CSRStructureModifyingFunctor<Device, T> { struct188 explicit CSRSparseMatrixAdd(OpKernelContext* ctx, const T alpha,
91 functor::CSRSparseMatrixAdd<Device, T> csr_geam(ctx_, alpha_, beta_); in operator ()()274 struct CSRSparseMatrixAdd<GPUDevice, T> struct276 explicit CSRSparseMatrixAdd(OpKernelContext* ctx, const T alpha, const T beta) in CSRSparseMatrixAdd() function383 TF_DISALLOW_COPY_AND_ASSIGN(CSRSparseMatrixAdd);