Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc62 Status VerifyReducerShape(const ProgramShape& reducer_shape, in VerifyReducerShape() argument
66 if (reducer_shape.parameters_size() != inputs * 2) { in VerifyReducerShape()
70 inputs * 2, reducer_shape.parameters_size()); in VerifyReducerShape()
73 const Shape& accumulator_shape = reducer_shape.result(); in VerifyReducerShape()
114 reducer_shape.parameters(i))) { in VerifyReducerShape()
118 i, ShapeUtil::HumanString(reducer_shape.parameters(i)), in VerifyReducerShape()
134 input_element_shape, reducer_shape.parameters(inputs + i))) { in VerifyReducerShape()
139 ShapeUtil::HumanString(reducer_shape.parameters(inputs + i)), in VerifyReducerShape()
148 *accumulator_subshapes[i], reducer_shape.parameters(inputs + i))) { in VerifyReducerShape()
153 ShapeUtil::HumanString(reducer_shape.parameters(inputs + i)), in VerifyReducerShape()