Searched refs:reducer_shape (Results 1 – 1 of 1) sorted by relevance
60 Status VerifyReducerShape(const ProgramShape& reducer_shape, in VerifyReducerShape() argument64 if (reducer_shape.parameters_size() != inputs * 2) { in VerifyReducerShape()68 inputs * 2, reducer_shape.parameters_size()); in VerifyReducerShape()71 const Shape& accumulator_shape = reducer_shape.result(); in VerifyReducerShape()112 reducer_shape.parameters(i))) { in VerifyReducerShape()116 i, ShapeUtil::HumanString(reducer_shape.parameters(i)), in VerifyReducerShape()132 input_element_shape, reducer_shape.parameters(inputs + i))) { in VerifyReducerShape()137 ShapeUtil::HumanString(reducer_shape.parameters(inputs + i)), in VerifyReducerShape()146 *accumulator_subshapes[i], reducer_shape.parameters(inputs + i))) { in VerifyReducerShape()151 ShapeUtil::HumanString(reducer_shape.parameters(inputs + i)), in VerifyReducerShape()