1 # 2 # Copyright (C) 2018 The Android Open Source Project 3 # 4 # Licensed under the Apache License, Version 2.0 (the "License"); 5 # you may not use this file except in compliance with the License. 6 # You may obtain a copy of the License at 7 # 8 # http://www.apache.org/licenses/LICENSE-2.0 9 # 10 # Unless required by applicable law or agreed to in writing, software 11 # distributed under the License is distributed on an "AS IS" BASIS, 12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 # See the License for the specific language governing permissions and 14 # limitations under the License. 15 # 16 17 input0 = Input("input0", "TENSOR_FLOAT32", "{1, 2, 3, 4, 5}") 18 output0 = Output("output0", "TENSOR_FLOAT32", "{1, 2, 3, 4, 5}") 19 model = Model().Operation("RSQRT", input0).To(output0) 20 21 input_data = [(i + 1) / 10 for i in range(120)] 22 output_data = [1.0 / math.sqrt(x) for x in input_data] 23 24 Example({ 25 input0: input_data, 26 output0: output_data, 27 }).AddVariations("relaxed", "float16") 28
