Searched refs:is_sqrtn (Results 1 – 1 of 1) sorted by relevance
622 bool is_mean, bool is_sqrtn, in SparseSegmentReductionOpBase() argument626 is_sqrtn_(is_sqrtn), in SparseSegmentReductionOpBase()1008 explicit SparseSegmentGradOpBase(OpKernelConstruction* context, bool is_sqrtn) in SparseSegmentGradOpBase() argument1009 : OpKernel(context), is_sqrtn_(is_sqrtn) {} in SparseSegmentGradOpBase()