| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/ |
| H A D | binary_op.py | 26 test_parameters=None): argument
29 if test_parameters is None:
30 test_parameters = []
32 test_parameters = test_parameters + [
158 test_parameters = test_parameters + [
170 test_parameters = test_parameters + [
188 test_parameters = [
189 test_parameter for test_parameter in test_parameters
237 test_parameters,
251 test_parameters = [
[all …]
|
| H A D | space_to_batch_nd.py | 28 test_parameters = [
106 test_parameters = [test_parameters[0], test_parameters[1], test_parameters[3]]
110 test_parameters,
|
| H A D | strided_slice.py | 25 def _make_strided_slice_tests(options, test_parameters, expected_tf_failures=0): argument
99 test_parameters,
110 test_parameters = [
296 _make_strided_slice_tests(options, test_parameters, expected_tf_failures=29)
302 test_parameters = [
317 _make_strided_slice_tests(options, test_parameters)
|
| H A D | shape_to_strided_slice.py | 23 test_parameters, argument
59 test_parameters,
69 test_parameters = [
97 options, test_parameters, expected_tf_failures=0)
|
| H A D | less_equal.py | 26 test_parameters = [{
41 test_parameters = test_parameters + [
84 test_parameters,
|
| H A D | elementwise.py | 28 test_parameters = [
44 test_parameters = [
45 test_parameter for test_parameter in test_parameters
66 make_zip_of_tests(options, test_parameters, build_graph, build_inputs)
|
| H A D | reshape.py | 28 test_parameters = [
61 test_parameters = test_parameters + [
107 make_zip_of_tests(options, test_parameters, build_graph, build_inputs)
|
| H A D | batch_to_space_nd.py | 27 test_parameters = [
61 test_parameters = test_parameters + [{
110 make_zip_of_tests(options, test_parameters, build_graph, build_inputs)
|
| H A D | pool.py | 44 test_parameters = [
80 test_parameters = [
81 test_parameter for test_parameter in test_parameters
107 test_parameters,
|
| H A D | fully_connected.py | 27 test_parameters = [{
126 test_parameters = test_parameters + [
183 test_parameters,
|
| /aosp_15_r20/external/executorch/backends/arm/test/ops/ |
| H A D | test_squeeze.py | 25 test_parameters: list[tuple[torch.Tensor, int]] = [ variable in TestSqueeze.SqueezeDim
35 test_parameters: list[tuple[torch.Tensor, tuple[int]]] = [ variable in TestSqueeze.SqueezeDims
45 test_parameters: list[tuple[torch.Tensor]] = [ variable in TestSqueeze.Squeeze
115 @parameterized.expand(Squeeze.test_parameters)
124 @parameterized.expand(Squeeze.test_parameters)
133 @parameterized.expand(Squeeze.test_parameters)
145 @parameterized.expand(Squeeze.test_parameters)
157 @parameterized.expand(SqueezeDim.test_parameters)
163 @parameterized.expand(SqueezeDim.test_parameters)
169 @parameterized.expand(SqueezeDim.test_parameters)
[all …]
|
| H A D | test_var.py | 30 test_parameters = [ variable in TestVar.Var
46 test_parameters = [ variable in TestVar.VarDim
63 test_parameters = [ variable in TestVar.VarCorrection
143 @parameterized.expand(Var.test_parameters)
147 @parameterized.expand(Var.test_parameters)
151 @parameterized.expand(Var.test_parameters)
159 @parameterized.expand(Var.test_parameters)
167 @parameterized.expand(VarDim.test_parameters)
173 @parameterized.expand(VarDim.test_parameters)
179 @parameterized.expand(VarDim.test_parameters)
[all …]
|
| H A D | test_bmm.py | 24 test_parameters = [ variable in TestBMM.BMM
36 test_parameters = [(torch.rand(2, 3, 5), torch.rand(2, 5, 2))] variable in TestBMM.MatMul
42 test_parameters = [ variable in TestBMM.BMMSingleInput
115 @parameterized.expand(BMM.test_parameters)
120 @parameterized.expand(BMMSingleInput.test_parameters)
125 @parameterized.expand(MatMul.test_parameters)
130 @parameterized.expand(MatMul.test_parameters)
135 @parameterized.expand(BMM.test_parameters)
140 @parameterized.expand(BMMSingleInput.test_parameters)
145 @parameterized.expand(BMM.test_parameters)
[all …]
|
| H A D | test_sub.py | 22 test_parameters = [ variable in TestSimpleSub.Sub
32 test_parameters = [ variable in TestSimpleSub.Sub2
101 @parameterized.expand(Sub.test_parameters)
106 @parameterized.expand(Sub.test_parameters)
111 @parameterized.expand(Sub.test_parameters)
118 @parameterized.expand(Sub.test_parameters)
125 @parameterized.expand(Sub2.test_parameters)
130 @parameterized.expand(Sub2.test_parameters)
135 @parameterized.expand(Sub2.test_parameters)
142 @parameterized.expand(Sub2.test_parameters)
|
| H A D | test_mm.py | 28 test_parameters = [ variable in TestMM.MM
40 test_parameters = [ variable in TestMM.MMSingleInput
113 @parameterized.expand(MM.test_parameters)
118 @parameterized.expand(MMSingleInput.test_parameters)
123 @parameterized.expand(MM.test_parameters)
128 @parameterized.expand(MMSingleInput.test_parameters)
134 @parameterized.expand(MM.test_parameters)
143 @parameterized.expand(MMSingleInput.test_parameters)
151 @parameterized.expand(MM.test_parameters)
158 @parameterized.expand(MMSingleInput.test_parameters)
|
| H A D | test_add.py | 24 test_parameters = [ variable in TestSimpleAdd.Add
36 test_parameters = [ variable in TestSimpleAdd.Add2
121 @parameterized.expand(Add.test_parameters)
126 @parameterized.expand(Add.test_parameters)
131 @parameterized.expand(Add.test_parameters)
144 @parameterized.expand(Add.test_parameters)
157 @parameterized.expand(Add2.test_parameters)
162 @parameterized.expand(Add2.test_parameters)
167 @parameterized.expand(Add2.test_parameters)
178 @parameterized.expand(Add2.test_parameters)
|
| H A D | test_rsqrt.py | 21 test_parameters = [ variable in TestRsqrt.Rsqrt
89 @parameterized.expand(Rsqrt.test_parameters)
93 @parameterized.expand(Rsqrt.test_parameters)
97 @parameterized.expand(Rsqrt.test_parameters)
103 @parameterized.expand(Rsqrt.test_parameters)
|
| H A D | test_unsqueeze.py | 26 test_parameters: list[tuple[torch.Tensor]] = [(torch.randn(n),) for n in shapes] variable in TestSimpleUnsqueeze.Unsqueeze
89 @parameterized.expand(Unsqueeze.test_parameters)
94 @parameterized.expand(Unsqueeze.test_parameters)
98 @parameterized.expand(Unsqueeze.test_parameters[:-1])
106 @parameterized.expand(Unsqueeze.test_parameters)
|
| H A D | test_expand.py | 33 test_parameters = [ variable in TestSimpleExpand.Expand
100 @parameterized.expand(Expand.test_parameters)
104 @parameterized.expand(Expand.test_parameters)
108 @parameterized.expand(Expand.test_parameters)
114 @parameterized.expand(Expand.test_parameters)
|
| H A D | test_repeat.py | 33 test_parameters = [ variable in TestSimpleRepeat.Repeat
101 @parameterized.expand(Repeat.test_parameters)
105 @parameterized.expand(Repeat.test_parameters)
109 @parameterized.expand(Repeat.test_parameters)
115 @parameterized.expand(Repeat.test_parameters)
|
| H A D | test_sum.py | 29 test_parameters: list[Tuple[exampledata_t]] = [ variable in TestSum.Sum
107 @parameterized.expand(Sum.test_parameters)
111 @parameterized.expand(Sum.test_parameters)
115 @parameterized.expand(Sum.test_parameters)
123 @parameterized.expand(Sum.test_parameters)
|
| H A D | test_clone.py | 34 test_parameters = [(torch.ones(n),) for n in sizes] variable in TestSimpleClone.Clone
113 @parameterized.expand(Clone.test_parameters)
117 @parameterized.expand(Clone.test_parameters)
121 @parameterized.expand(Clone.test_parameters)
125 @parameterized.expand(Clone.test_parameters)
|
| H A D | test_cat.py | 23 test_parameters = [ variable in TestCat.Cat
116 @parameterized.expand(Cat.test_parameters)
127 @parameterized.expand(Cat.test_parameters)
132 @parameterized.expand(Cat.test_parameters)
139 @parameterized.expand(Cat.test_parameters)
|
| H A D | test_full.py | 42 test_parameters = [((torch.randn(n) * 10 - 5, 3.2),) for n in sizes] variable in TestFull.AddVariableFull
133 @parameterized.expand(AddVariableFull.test_parameters)
139 @parameterized.expand(AddVariableFull.test_parameters)
143 @parameterized.expand(AddVariableFull.test_parameters)
150 @parameterized.expand(AddVariableFull.test_parameters)
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/ |
| H A D | shape_to_strided_slice.py | 27 test_parameters, argument
63 test_parameters,
73 test_parameters = [
101 options, test_parameters, expected_tf_failures=0)
|