Searched refs:AssignSub (Results 1 – 20 of 20) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | fused_batch_norm_fusion_test.py | 23 AssignSub = P.AssignSub() variable
59 assign_sub0 = AssignSub(var0, mul0, U)
61 assign_sub1 = AssignSub(var1, mul1, u0)
76 assign_sub0 = AssignSub(var0, Cast(mul0, mstype.float32), U)
78 assign_sub1 = AssignSub(var1, Cast(mul1, mstype.float32), u0)
93 assign_sub0 = AssignSub(var0, mul0, U)
95 assign_sub1 = AssignSub(var1, mul1, u0)
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_AssignSub.py | 32 self.AssignSub = P.AssignSub()
37 out = self.AssignSub(self.inputdata, x)
|
| /third_party/mindspore/tests/syntax/simple_expression/ |
| D | test_assignment_ops.py | 267 class AssignSub(nn.Cell): class
269 super(AssignSub, self).__init__()
272 self.assignsub = P.AssignSub()
284 result2 = AssignSub(result2, input_x)()
295 result2 = AssignSub(result2, input_x)()
306 result2 = AssignSub(result2, input_x)()
317 result2 = AssignSub(result2, input_x)()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | elewise_calculation_ops_declare.h | 51 DECLARE_OP_ADAPTER(AssignSub)
52 DECLARE_OP_USE_OUTPUT(AssignSub)
|
| D | elewise_calculation_ops_declare.cc | 100 INPUT_MAP(AssignSub) = {{1, INPUT_DESC(var)}, {2, INPUT_DESC(value)}};
101 ATTR_MAP(AssignSub) = EMPTY_ATTR_MAP;
102 OUTPUT_MAP(AssignSub) = {{0, OUTPUT_DESC(var)}};
103 REG_ADPT_DESC(AssignSub, kNameAssignSub, ADPT_DESC(AssignSub))
|
| /third_party/mindspore/tests/ut/python/exec/ |
| D | test_assign_sub.py | 34 self.sub = P.AssignSub()
|
| /third_party/mindspore/mindspore/ops/bprop_mindir/ |
| D | generate_mindir.py | 26 … P.Broadcast(1), P.AssignAdd(), P.AssignSub(), P.IsFinite(), P.ApproximateEqual(), P.Sign(),
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_prelu_cell.py | 75 self.assign_sub = P.AssignSub().shard(((1,), (1,)))
|
| D | test_arithmetic.py | 619 self.assign_sub = P.AssignSub()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 49 from .math_ops import (Abs, ACos, Asin, Asinh, AddN, AccumulateNV2, AssignAdd, AssignSub, Atan2, Ba…
|
| D | math_ops.py | 295 class AssignSub(PrimitiveWithInfer): class
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | normalization.py | 159 self.assign_sub_mean = P.AssignSub().shard(data_parallel_strategy)
160 self.assign_sub_var = P.AssignSub().shard(data_parallel_strategy)
|
| D | quant.py | 770 self.assign_sub_mean = P.AssignSub()
771 self.assign_sub_var = P.AssignSub()
|
| /third_party/mindspore/mindspore/ops/ |
| D | functional.py | 85 assign_sub = P.AssignSub()
|
| /third_party/mindspore/tests/ut/python/optimizer/ |
| D | test_bprop_mindir.py | 240 self.assign = P.AssignSub()
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_effect_ops.py | 60 self.assign_sub = P.AssignSub()
|
| D | test_auto_monad.py | 834 self.assign_sub = P.AssignSub()
1273 self.assign_sub = P.AssignSub()
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_math_ops_check.py | 44 self.op = P.AssignSub()
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_math_ops.py | 946 @bprop_getters.register(P.AssignSub)
|
| /third_party/mindspore/config/ |
| D | op_info.config | 183 {"op_name": "AssignSub", "inputs": [{"index": 0, "name": "var", "need_compile": false, "param_type"…
|