Home
last modified time | relevance | path

Searched refs:OpNetWrapper (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/tests/st/ops/cpu/
Dtest_split_op.py27 class OpNetWrapper(nn.Cell): class
29 super(OpNetWrapper, self).__init__()
41 op_wrapper = OpNetWrapper(op)
56 op_wrapper = OpNetWrapper(op)
73 op_wrapper = OpNetWrapper(op)
88 op_wrapper = OpNetWrapper(op)
97 op_wrapper = OpNetWrapper(op)
110 op_wrapper = OpNetWrapper(op)
119 op_wrapper = OpNetWrapper(op)
132 op_wrapper = OpNetWrapper(op)
[all …]
Dtest_gathernd_op.py28 class OpNetWrapper(nn.Cell): class
30 super(OpNetWrapper, self).__init__()
42 op_wrapper = OpNetWrapper(op)
57 op_wrapper = OpNetWrapper(op)
72 op_wrapper = OpNetWrapper(op)
88 op_wrapper = OpNetWrapper(op)
104 op_wrapper = OpNetWrapper(op)
120 op_wrapper = OpNetWrapper(op)
136 op_wrapper = OpNetWrapper(op)
152 op_wrapper = OpNetWrapper(op)
[all …]
Dtest_logical_op.py27 class OpNetWrapper(nn.Cell): class
29 super(OpNetWrapper, self).__init__()
41 op_wrapper = OpNetWrapper(op)
55 op_wrapper = OpNetWrapper(op)
69 op_wrapper = OpNetWrapper(op)
Dtest_range_op.py25 class OpNetWrapper(nn.Cell): class
27 super(OpNetWrapper, self).__init__()
39 op_wrapper = OpNetWrapper(op)
52 op_wrapper = OpNetWrapper(op)
Dtest_sign_op.py27 class OpNetWrapper(nn.Cell): class
29 super(OpNetWrapper, self).__init__()
41 op_wrapper = OpNetWrapper(op)
55 op_wrapper = OpNetWrapper(op)
Dtest_notequal_op.py27 class OpNetWrapper(nn.Cell): class
29 super(OpNetWrapper, self).__init__()
41 op_wrapper = OpNetWrapper(op)
56 op_wrapper = OpNetWrapper(op)
Dtest_greater_op.py27 class OpNetWrapper(nn.Cell): class
29 super(OpNetWrapper, self).__init__()
41 op_wrapper = OpNetWrapper(op)
57 op_wrapper = OpNetWrapper(op)
Dtest_greater_equal_op.py27 class OpNetWrapper(nn.Cell): class
29 super(OpNetWrapper, self).__init__()
41 op_wrapper = OpNetWrapper(op)
57 op_wrapper = OpNetWrapper(op)