Searched refs:GroupShape (Results 1 – 1 of 1) sorted by relevance
53 Status GroupShape(const VarDimArray& input_shape, ShapeArray* grouped_shape) { in GroupShape() function259 OP_REQUIRES_OK(ctx, GroupShape(set_st.shape(), &output_shape)); in Compute()394 TF_RETURN_IF_ERROR(GroupShape(shape1, &group_shape_1)); in GroupShapeFromInputs()396 TF_RETURN_IF_ERROR(GroupShape(shape2, &group_shape_2)); in GroupShapeFromInputs()