1# Copyright 2023 Huawei Technologies Co., Ltd 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14# ============================================================================ 15""" test graph fallback """ 16from mindspore import ops 17import mindspore.ops.operations as P 18 19 20def add_func(x, y): 21 return x + y 22 23 24add = ops.MultitypeFuncGraph('add1') 25tensor_add = ops.Add() 26 27 28@add.register("Number", "Number") 29def add_scala(x, y): 30 return x + y 31 32 33@add.register("Tensor", "Tensor") 34def add_tensor(x, y): 35 return tensor_add(x, y) 36 37 38c = ops.MultitypeFuncGraph('concat') 39 40 41@c.register('List') 42def concat(t): 43 return P.Concat()(t) 44