Searched refs:accumulator_subshapes (Results 1 – 1 of 1) sorted by relevance
72 std::vector<const Shape*> accumulator_subshapes; in VerifyReducerShape() local80 accumulator_subshapes.push_back(&accumulator_shape); in VerifyReducerShape()89 accumulator_subshapes.push_back(&element_shape); in VerifyReducerShape()98 for (const Shape* element_shape : accumulator_subshapes) { in VerifyReducerShape()111 if (!ShapeUtil::Compatible(*accumulator_subshapes[i], in VerifyReducerShape()117 ShapeUtil::HumanString(*accumulator_subshapes[i])); in VerifyReducerShape()120 if (!ShapeUtil::CompatibleIgnoringFpPrecision(*accumulator_subshapes[i], in VerifyReducerShape()125 i, ShapeUtil::HumanString(*accumulator_subshapes[i]), in VerifyReducerShape()146 *accumulator_subshapes[i], reducer_shape.parameters(inputs + i))) { in VerifyReducerShape()152 ShapeUtil::HumanString(*accumulator_subshapes[i])); in VerifyReducerShape()