Searched refs:reducer_shape (Results 1 – 1 of 1) sorted by relevance
62 Status VerifyReducerShape(const ProgramShape& reducer_shape, in VerifyReducerShape() argument66 if (reducer_shape.parameters_size() != inputs * 2) { in VerifyReducerShape()70 inputs * 2, reducer_shape.parameters_size()); in VerifyReducerShape()73 const Shape& accumulator_shape = reducer_shape.result(); in VerifyReducerShape()114 reducer_shape.parameters(i))) { in VerifyReducerShape()118 i, ShapeUtil::HumanString(reducer_shape.parameters(i)), in VerifyReducerShape()134 input_element_shape, reducer_shape.parameters(inputs + i))) { in VerifyReducerShape()139 ShapeUtil::HumanString(reducer_shape.parameters(inputs + i)), in VerifyReducerShape()148 *accumulator_subshapes[i], reducer_shape.parameters(inputs + i))) { in VerifyReducerShape()153 ShapeUtil::HumanString(reducer_shape.parameters(inputs + i)), in VerifyReducerShape()