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 def test(input0, output0, axes, keep_dims, input_data, output_data): 18 model = Model().Operation("REDUCE_ALL", input0, axes, keep_dims).To(output0) 19 Example({ 20 input0: input_data, 21 output0: output_data, 22 }, model=model) 23 24 test( 25 input0=Input("input0", "TENSOR_BOOL8", "{1}"), 26 input_data=[False], 27 axes=[0], 28 keep_dims=True, 29 output0=Output("output0", "TENSOR_BOOL8", "{1}"), 30 output_data=[False], 31 ) 32 33 test( 34 input0=Input("input0", "TENSOR_BOOL8", "{2, 3, 2}"), 35 input_data=[True, True, True, True, True, False, 36 True, True, True, True, True, True], 37 axes=[1, 0, -3, -3], 38 keep_dims=False, 39 output0=Output("output0", "TENSOR_BOOL8", "{2}"), 40 output_data=[True, False], 41 ) 42 43 test( 44 input0=Input("input0", "TENSOR_BOOL8", "{2, 3, 2}"), 45 input_data=[True, True, True, True, True, True, 46 True, True, False, True, True, True], 47 axes=[0, 2], 48 keep_dims=True, 49 output0=Output("output0", "TENSOR_BOOL8", "{1, 3, 1}"), 50 output_data=[True, False, True], 51 ) 52
