Searched refs:SparseSegmentGradOpBase (Results 1 – 1 of 1) sorted by relevance
1006 class SparseSegmentGradOpBase : public OpKernel { class1008 explicit SparseSegmentGradOpBase(OpKernelConstruction* context, bool is_sqrtn) in SparseSegmentGradOpBase() function in tensorflow::SparseSegmentGradOpBase1112 class SparseSegmentMeanGradOp : public SparseSegmentGradOpBase<T> {1115 : SparseSegmentGradOpBase<T>(context, false /*is_sqrtn*/) {} in SparseSegmentMeanGradOp()1119 class SparseSegmentSqrtNGradOp : public SparseSegmentGradOpBase<T> {1122 : SparseSegmentGradOpBase<T>(context, true /*is_sqrtn*/) {} in SparseSegmentSqrtNGradOp()