Searched refs:accumulator_subshapes (Results 1 – 1 of 1) sorted by relevance
74 std::vector<const Shape*> accumulator_subshapes; in VerifyReducerShape() local82 accumulator_subshapes.push_back(&accumulator_shape); in VerifyReducerShape()91 accumulator_subshapes.push_back(&element_shape); in VerifyReducerShape()100 for (const Shape* element_shape : accumulator_subshapes) { in VerifyReducerShape()113 if (!ShapeUtil::Compatible(*accumulator_subshapes[i], in VerifyReducerShape()119 ShapeUtil::HumanString(*accumulator_subshapes[i])); in VerifyReducerShape()122 if (!ShapeUtil::CompatibleIgnoringFpPrecision(*accumulator_subshapes[i], in VerifyReducerShape()127 i, ShapeUtil::HumanString(*accumulator_subshapes[i]), in VerifyReducerShape()148 *accumulator_subshapes[i], reducer_shape.parameters(inputs + i))) { in VerifyReducerShape()154 ShapeUtil::HumanString(*accumulator_subshapes[i])); in VerifyReducerShape()