Searched refs:accumulator_shape (Results 1 – 1 of 1) sorted by relevance
73 const Shape& accumulator_shape = reducer_shape.result(); in VerifyReducerShape() local75 if (accumulator_shape.IsArray()) { in VerifyReducerShape()82 accumulator_subshapes.push_back(&accumulator_shape); in VerifyReducerShape()83 } else if (accumulator_shape.IsTuple()) { in VerifyReducerShape()84 if (ShapeUtil::TupleElementCount(accumulator_shape) != inputs) { in VerifyReducerShape()88 inputs, ShapeUtil::TupleElementCount(accumulator_shape)); in VerifyReducerShape()90 for (const Shape& element_shape : accumulator_shape.tuple_shapes()) { in VerifyReducerShape()97 ShapeUtil::HumanString(accumulator_shape)); in VerifyReducerShape()105 ShapeUtil::HumanString(accumulator_shape)); in VerifyReducerShape()