Searched refs:num_output_elems (Results 1 – 1 of 1) sorted by relevance
3662 int64 num_output_elems = ShapeUtil::ElementsIn(first_reduce->shape()); in ComputeMappingSchemeAndReductionKind() local3666 CHECK_EQ(num_output_elems, num_kept); in ComputeMappingSchemeAndReductionKind()