Searched refs:VerifyReducerShape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 60 Status VerifyReducerShape(const ProgramShape& reducer_shape, in VerifyReducerShape() function
2085 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply, init_values, element_types, in InferReduceShape()
2116 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, {&init_value_shape}, in InferReduceWindowShape()
2163 TF_RETURN_IF_ERROR(VerifyReducerShape(scatter_shape, {&init_value_shape}, in InferSelectAndScatterShape()
3193 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, {&init_value_shape}, in InferScatterShape()
|