Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc2126 auto reduced_args = arg_shapes.subspan(0, num_reduced_args); in InferReduceShape() local
2130 if (!ShapeUtil::SameDimensions(*reduced_args[0], *reduced_args[i])) { in InferReduceShape()
2134 ShapeUtil::HumanString(*reduced_args[0]), i, in InferReduceShape()
2135 ShapeUtil::HumanString(*reduced_args[i])); in InferReduceShape()
2141 const Shape& arg = *reduced_args[0]; in InferReduceShape()
2151 for (const Shape* arg : reduced_args) { in InferReduceShape()