Searched refs:AxisConverter (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/tools/test_generator/ |
| D | README.md | 118 - AxisConverter
145 #### AxisConverter subsubsection
150 converter = AxisConverter(originalAxis, targetAxis, dimension, drop=[], name="variation_name").Iden…
158 toAxis0 = AxisConverter(-1, 0, 4).Identify([input, output, axis])
164 toAxis0 = AxisConverter(-1, -4, 4).Identify([input, output, axis])
170 toDim2 = AxisConverter(-1, -1, 4, drop=[0, 1]).Identify([input, output, axis])
176 toDim2Axis0 = AxisConverter(-1, 2, 4, drop=[0, 1]).Identify([input, output, axis])
269 # AxisConverter
271 example.AddVariations(*[AxisConverter(origin, t, dim).Identify(op_list) for t in targets])
275 AxisConverter(origin, t, dim).Identify(op_list) for t in range(dim)
[all …]
|
| D | test_generator.py | 857 class AxisConverter(ModelVariation): class
1149 var = [AxisConverter(origin, a, dim).Identify(args) for a in axis]
1155 var = [AxisConverter(origin, a, dim).Identify(args) for a in range(dim)]
1161 var = [AxisConverter(origin, a, dim).Identify(args) for a in range(-dim, dim)]
1170 var = [AxisConverter(origin, origin, dim, drop[0:(dim-i)]).Identify(args) for i in dims]
1178 var = [AxisConverter(origin, origin, dim, drop[0:i]).Identify(args) for i in range(dim)]
1184 var = [AxisConverter(origin, j, dim, range(i)).Identify(args) \
1191 var = [AxisConverter(origin, k, dim, range(i)).Identify(args) \
|