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