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()
|