Searched refs:AddAllDimsAndAxis (Results 1 – 9 of 9) sorted by relevance
| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | l2_normalization_axis.mod.py | 48 Example(example0).AddAllDimsAndAxis(i1, o1, axis).AddVariations("relaxed", "float16", quant8)
60 }).AddAllDimsAndAxis(i2, o2, axis)
|
| D | local_response_normalization_v1_2.mod.py | 26 })).AddAllDimsAndAxis(i, o, axis).AddVariations("relaxed", "float16")
32 })).AddAllDimsAndAxis(i, o, axis).AddVariations("relaxed", "float16")
|
| D | softmax_v1_2.mod.py | 52 Example(example1).AddVariations("relaxed", "float16", quant8).AddAllDimsAndAxis(i, o, axis)
55 Example(example2).AddVariations("relaxed", "float16", quant8).AddAllDimsAndAxis(i, o, axis)
|
| D | channel_shuffle.mod.py | 42 }).AddVariations("relaxed", quant8, "float16").AddAllDimsAndAxis(i1, o1, axis)
|
| /frameworks/ml/nn/runtime/test/specs/V1_3/ |
| D | softmax_quant8_signed.mod.py | 94 Example(example1).AddVariations(quant8_signed, includeDefault=False).AddAllDimsAndAxis(i, o, axis)
97 Example(example2).AddVariations(quant8_signed, includeDefault=False).AddAllDimsAndAxis(i, o, axis)
|
| D | channel_shuffle_quant8_signed.mod.py | 42 }).AddVariations(quant8_signed, includeDefault=False).AddAllDimsAndAxis(i1, o1, axis)
|
| D | l2_normalization_quant8_signed.mod.py | 47 Example(example0).AddAllDimsAndAxis(i1, o1, axis).AddVariations(quant8_signed, includeDefault=False)
|
| /frameworks/ml/nn/tools/test_generator/ |
| D | README.md | 304 example.AddAllDimsAndAxis(dims, *op_list)
|
| D | test_generator.py | 1189 def AddAllDimsAndAxis(self, *args): member in Example
|