Searched refs:accumulator_shape (Results 1 – 1 of 1) sorted by relevance
71 const Shape& accumulator_shape = reducer_shape.result(); in VerifyReducerShape() local73 if (accumulator_shape.IsArray()) { in VerifyReducerShape()80 accumulator_subshapes.push_back(&accumulator_shape); in VerifyReducerShape()81 } else if (accumulator_shape.IsTuple()) { in VerifyReducerShape()82 if (ShapeUtil::TupleElementCount(accumulator_shape) != inputs) { in VerifyReducerShape()86 inputs, ShapeUtil::TupleElementCount(accumulator_shape)); in VerifyReducerShape()88 for (const Shape& element_shape : accumulator_shape.tuple_shapes()) { in VerifyReducerShape()95 ShapeUtil::HumanString(accumulator_shape)); in VerifyReducerShape()103 ShapeUtil::HumanString(accumulator_shape)); in VerifyReducerShape()