Home
last modified time | relevance | path

Searched refs:accumulator_shape (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc73 const Shape& accumulator_shape = reducer_shape.result(); in VerifyReducerShape() local
75 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()