1420ebdd2785f2545b0168688582635359486d6f7Yang Ni# 2420ebdd2785f2545b0168688582635359486d6f7Yang Ni# Copyright (C) 2017 The Android Open Source Project 3420ebdd2785f2545b0168688582635359486d6f7Yang Ni# 4420ebdd2785f2545b0168688582635359486d6f7Yang Ni# Licensed under the Apache License, Version 2.0 (the "License"); 5420ebdd2785f2545b0168688582635359486d6f7Yang Ni# you may not use this file except in compliance with the License. 6420ebdd2785f2545b0168688582635359486d6f7Yang Ni# You may obtain a copy of the License at 7420ebdd2785f2545b0168688582635359486d6f7Yang Ni# 8420ebdd2785f2545b0168688582635359486d6f7Yang Ni# http://www.apache.org/licenses/LICENSE-2.0 9420ebdd2785f2545b0168688582635359486d6f7Yang Ni# 10420ebdd2785f2545b0168688582635359486d6f7Yang Ni# Unless required by applicable law or agreed to in writing, software 11420ebdd2785f2545b0168688582635359486d6f7Yang Ni# distributed under the License is distributed on an "AS IS" BASIS, 12420ebdd2785f2545b0168688582635359486d6f7Yang Ni# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13420ebdd2785f2545b0168688582635359486d6f7Yang Ni# See the License for the specific language governing permissions and 14420ebdd2785f2545b0168688582635359486d6f7Yang Ni# limitations under the License. 15420ebdd2785f2545b0168688582635359486d6f7Yang Ni# 16420ebdd2785f2545b0168688582635359486d6f7Yang Ni 17420ebdd2785f2545b0168688582635359486d6f7Yang Nibatches = 2 188db6e487763be346310fc5468bd8f33f255ccb9cYang Nifeatures = 4 198db6e487763be346310fc5468bd8f33f255ccb9cYang Nirank = 1 208db6e487763be346310fc5468bd8f33f255ccb9cYang Niunits = int(features / rank) 21420ebdd2785f2545b0168688582635359486d6f7Yang Niinput_size = 3 22420ebdd2785f2545b0168688582635359486d6f7Yang Nimemory_size = 10 23420ebdd2785f2545b0168688582635359486d6f7Yang Ni 24420ebdd2785f2545b0168688582635359486d6f7Yang Nimodel = Model() 25420ebdd2785f2545b0168688582635359486d6f7Yang Ni 26420ebdd2785f2545b0168688582635359486d6f7Yang Niinput = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (batches, input_size)) 278db6e487763be346310fc5468bd8f33f255ccb9cYang Niweights_feature = Input("weights_feature", "TENSOR_FLOAT32", "{%d, %d}" % (features, input_size)) 288db6e487763be346310fc5468bd8f33f255ccb9cYang Niweights_time = Input("weights_time", "TENSOR_FLOAT32", "{%d, %d}" % (features, memory_size)) 29420ebdd2785f2545b0168688582635359486d6f7Yang Nibias = Input("bias", "TENSOR_FLOAT32", "{%d}" % (units)) 308db6e487763be346310fc5468bd8f33f255ccb9cYang Nistate_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*features)) 31e538bb0f9db55531dbc018c6b81ff1d6e4fbd8acMichael Butlerrank_param = Int32Scalar("rank_param", rank) 32e538bb0f9db55531dbc018c6b81ff1d6e4fbd8acMichael Butleractivation_param = Int32Scalar("activation_param", 0) 338db6e487763be346310fc5468bd8f33f255ccb9cYang Nistate_out = IgnoredOutput("state_out", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*features)) 34420ebdd2785f2545b0168688582635359486d6f7Yang Nioutput = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (batches, units)) 35420ebdd2785f2545b0168688582635359486d6f7Yang Ni 36eefb1e60444afd08a4350e11f281ac7064ebba63Yang Nimodel = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in, 37eefb1e60444afd08a4350e11f281ac7064ebba63Yang Ni rank_param, activation_param).To([state_out, output]) 38420ebdd2785f2545b0168688582635359486d6f7Yang Ni 39420ebdd2785f2545b0168688582635359486d6f7Yang Niinput0 = { 408db6e487763be346310fc5468bd8f33f255ccb9cYang Ni input: [], 41420ebdd2785f2545b0168688582635359486d6f7Yang Ni weights_feature: [ 42420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.31930989, -0.36118156, 0.0079667, 0.37613347, 43420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.22197971, 0.12416199, 0.27901134, 0.27557442, 44420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.3905206, -0.36137494, -0.06634006, -0.10640851 45420ebdd2785f2545b0168688582635359486d6f7Yang Ni ], 46420ebdd2785f2545b0168688582635359486d6f7Yang Ni weights_time: [ 47420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.31930989, 0.37613347, 0.27901134, -0.36137494, -0.36118156, 48420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.22197971, 0.27557442, -0.06634006, 0.0079667, 0.12416199, 49420ebdd2785f2545b0168688582635359486d6f7Yang Ni 50420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.3905206, -0.10640851, -0.0976817, 0.15294972, 0.39635518, 51420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.02702999, 0.39296314, 0.15785322, 0.21931258, 0.31053296, 52420ebdd2785f2545b0168688582635359486d6f7Yang Ni 53420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.36916667, 0.38031587, -0.21580373, 0.27072677, 0.23622236, 54420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.34936687, 0.18174365, 0.35907319, -0.17493086, 0.324846, 55420ebdd2785f2545b0168688582635359486d6f7Yang Ni 56420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.10781813, 0.27201805, 0.14324132, -0.23681851, -0.27115166, 57420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.01580888, -0.14943552, 0.15465137, 0.09784451, -0.0337657 58420ebdd2785f2545b0168688582635359486d6f7Yang Ni ], 59420ebdd2785f2545b0168688582635359486d6f7Yang Ni bias: [], 608db6e487763be346310fc5468bd8f33f255ccb9cYang Ni state_in: [0 for _ in range(batches * memory_size * features)], 61420ebdd2785f2545b0168688582635359486d6f7Yang Ni} 62420ebdd2785f2545b0168688582635359486d6f7Yang Ni 63420ebdd2785f2545b0168688582635359486d6f7Yang Nitest_inputs = [ 64420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.12609188, -0.46347019, -0.89598465, 65420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.12609188, -0.46347019, -0.89598465, 66420ebdd2785f2545b0168688582635359486d6f7Yang Ni 67420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.14278367, -1.64410412, -0.75222826, 68420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.14278367, -1.64410412, -0.75222826, 69420ebdd2785f2545b0168688582635359486d6f7Yang Ni 70420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.49837467, 0.19278903, 0.26584083, 71420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.49837467, 0.19278903, 0.26584083, 72420ebdd2785f2545b0168688582635359486d6f7Yang Ni 73420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.11186574, 0.13164264, -0.05349274, 74420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.11186574, 0.13164264, -0.05349274, 75420ebdd2785f2545b0168688582635359486d6f7Yang Ni 76420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.68892461, 0.37783599, 0.18263303, 77420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.68892461, 0.37783599, 0.18263303, 78420ebdd2785f2545b0168688582635359486d6f7Yang Ni 79420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.81299269, -0.86831826, 1.43940818, 80420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.81299269, -0.86831826, 1.43940818, 81420ebdd2785f2545b0168688582635359486d6f7Yang Ni 82420ebdd2785f2545b0168688582635359486d6f7Yang Ni -1.45006323, -0.82251364, -1.69082689, 83420ebdd2785f2545b0168688582635359486d6f7Yang Ni -1.45006323, -0.82251364, -1.69082689, 84420ebdd2785f2545b0168688582635359486d6f7Yang Ni 85420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.03966608, -0.24936394, -0.77526885, 86420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.03966608, -0.24936394, -0.77526885, 87420ebdd2785f2545b0168688582635359486d6f7Yang Ni 88420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.11771342, -0.23761693, -0.65898693, 89420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.11771342, -0.23761693, -0.65898693, 90420ebdd2785f2545b0168688582635359486d6f7Yang Ni 91420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.89477462, 1.67204106, -0.53235275, 92420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.89477462, 1.67204106, -0.53235275 93420ebdd2785f2545b0168688582635359486d6f7Yang Ni] 94420ebdd2785f2545b0168688582635359486d6f7Yang Ni 95420ebdd2785f2545b0168688582635359486d6f7Yang Nigolden_outputs = [ 96420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.014899, -0.0517661, -0.143725, -0.00271883, 97420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.014899, -0.0517661, -0.143725, -0.00271883, 98420ebdd2785f2545b0168688582635359486d6f7Yang Ni 99420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.068281, -0.162217, -0.152268, 0.00323521, 100420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.068281, -0.162217, -0.152268, 0.00323521, 101420ebdd2785f2545b0168688582635359486d6f7Yang Ni 102420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.0317821, -0.0333089, 0.0609602, 0.0333759, 103420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.0317821, -0.0333089, 0.0609602, 0.0333759, 104420ebdd2785f2545b0168688582635359486d6f7Yang Ni 105420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.00623099, -0.077701, -0.391193, -0.0136691, 106420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.00623099, -0.077701, -0.391193, -0.0136691, 107420ebdd2785f2545b0168688582635359486d6f7Yang Ni 108420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.201551, -0.164607, -0.179462, -0.0592739, 109420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.201551, -0.164607, -0.179462, -0.0592739, 110420ebdd2785f2545b0168688582635359486d6f7Yang Ni 111420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.0886511, -0.0875401, -0.269283, 0.0281379, 112420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.0886511, -0.0875401, -0.269283, 0.0281379, 113420ebdd2785f2545b0168688582635359486d6f7Yang Ni 114420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.201174, -0.586145, -0.628624, -0.0330412, 115420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.201174, -0.586145, -0.628624, -0.0330412, 116420ebdd2785f2545b0168688582635359486d6f7Yang Ni 117420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.0839096, -0.299329, 0.108746, 0.109808, 118420ebdd2785f2545b0168688582635359486d6f7Yang Ni -0.0839096, -0.299329, 0.108746, 0.109808, 119420ebdd2785f2545b0168688582635359486d6f7Yang Ni 120420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.419114, -0.237824, -0.422627, 0.175115, 121420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.419114, -0.237824, -0.422627, 0.175115, 122420ebdd2785f2545b0168688582635359486d6f7Yang Ni 123420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.36726, -0.522303, -0.456502, -0.175475, 124420ebdd2785f2545b0168688582635359486d6f7Yang Ni 0.36726, -0.522303, -0.456502, -0.175475 125420ebdd2785f2545b0168688582635359486d6f7Yang Ni] 126420ebdd2785f2545b0168688582635359486d6f7Yang Ni 1278db6e487763be346310fc5468bd8f33f255ccb9cYang Nioutput0 = {state_out: [0 for _ in range(batches * memory_size * features)], 1288db6e487763be346310fc5468bd8f33f255ccb9cYang Ni output: []} 129420ebdd2785f2545b0168688582635359486d6f7Yang Ni 130420ebdd2785f2545b0168688582635359486d6f7Yang Ni# TODO: enable more data points after fixing the reference issue 131420ebdd2785f2545b0168688582635359486d6f7Yang Nifor i in range(1): 132420ebdd2785f2545b0168688582635359486d6f7Yang Ni batch_start = i * input_size * batches 133420ebdd2785f2545b0168688582635359486d6f7Yang Ni batch_end = batch_start + input_size * batches 134420ebdd2785f2545b0168688582635359486d6f7Yang Ni input0[input] = test_inputs[batch_start:batch_end] 135420ebdd2785f2545b0168688582635359486d6f7Yang Ni golden_start = i * units * batches 136420ebdd2785f2545b0168688582635359486d6f7Yang Ni golden_end = golden_start + units * batches 137420ebdd2785f2545b0168688582635359486d6f7Yang Ni output0[output] = golden_outputs[golden_start:golden_end] 138420ebdd2785f2545b0168688582635359486d6f7Yang Ni Example((input0, output0)) 139