| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_assign_add_op.py | 25 class AssignAdd(nn.Cell): class
27 super(AssignAdd, self).__init__()
29 self.add = P.AssignAdd()
65 add = AssignAdd(x1)
68 add = AssignAdd(output1)
73 add = AssignAdd(x2)
76 add = AssignAdd(output1)
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | assign_add.h | 31 class MS_CORE_API AssignAdd : public PrimitiveC {
34 AssignAdd() : PrimitiveC(kNameAssignAdd) { InitIOName({"ref", "value"}, {"output"}); } in AssignAdd() function
36 ~AssignAdd() = default;
37 MS_DECLARE_PARENT(AssignAdd, PrimitiveC);
43 using PrimAssignAddPtr = std::shared_ptr<AssignAdd>;
|
| D | assign_add.cc | 46 REGISTER_PRIMITIVE_C(kNameAssignAdd, AssignAdd);
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_assign_add_op.py | 25 class AssignAdd(nn.Cell): class
27 super(AssignAdd, self).__init__()
29 self.add = P.AssignAdd()
63 add = AssignAdd(x2)
66 add = AssignAdd(output1)
|
| /third_party/mindspore/tests/ut/python/exec/ |
| D | test_AssignAdd.py | 36 self.AssignAdd = P.AssignAdd()
41 out = self.AssignAdd(self.inputdata, x)
88 self.AssignAdd = P.AssignAdd()
93 z1 = self.AssignAdd(self.inputdata, self.one)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | assign_add_gpu_kernel.cc | 22 …AssignAdd, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAtt…
25 …AssignAdd, KernelAttr().AddInputAttr(kNumberTypeInt64).AddInputAttr(kNumberTypeInt64).AddOutputAtt…
28 AssignAdd,
32 AssignAdd,
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_assign_add.py | 25 class AssignAdd(nn.Cell): class
27 super(AssignAdd, self).__init__()
29 self.add = P.AssignAdd()
37 add = AssignAdd(x2)
39 add_2 = AssignAdd(result_gk_on_1)
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_AssignAdd.py | 32 self.AssignAdd = P.AssignAdd()
37 out = self.AssignAdd(self.inputdata, x)
|
| /third_party/mindspore/tests/syntax/simple_expression/ |
| D | test_assignment_ops.py | 188 class AssignAdd(nn.Cell): class
190 super(AssignAdd, self).__init__()
193 self.assignadd = P.AssignAdd()
205 assignadd = AssignAdd(result2, input_x)
217 result2 = AssignAdd(result2, input_x)()
228 result2 = AssignAdd(result2, input_x)()
239 result2 = AssignAdd(result2, input_x)()
250 result2 = AssignAdd(result2, input_x)()
261 result2 = AssignAdd(result2, input_x)()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | assign_add_impl.cu | 21 __global__ void AssignAdd(const size_t size, T* ref, const T* value, T* output) { in AssignAdd() function
31 AssignAdd<<<GET_BLOCKS(size), GET_THREADS, 0, cuda_stream>>>(size, ref, value, output); in CalAssignAdd()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | arithmetic_cpu_kernel.h | 50 void AssignAdd(T *input1, const T *input2, T *out) const;
116 …AssignAdd, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAtt…
119 …AssignAdd, KernelAttr().AddInputAttr(kNumberTypeInt64).AddInputAttr(kNumberTypeInt64).AddOutputAtt…
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/expanders/ |
| D | assign_add.py | 20 class AssignAdd(Expander): class
|
| D | __init__.py | 18 from .assign_add import AssignAdd
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | assignadd_cpu_kernel.h | 37 AssignAdd,
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | assign_add_infer.c | 38 REG_INFER(AssignAdd, PrimType_AssignAdd, AssignAddInferShape)
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | elewise_calculation_ops_declare.h | 48 DECLARE_OP_ADAPTER(AssignAdd)
49 DECLARE_OP_USE_OUTPUT(AssignAdd)
|
| D | elewise_calculation_ops_declare.cc | 94 INPUT_MAP(AssignAdd) = {{1, INPUT_DESC(ref)}, {2, INPUT_DESC(value)}};
95 ATTR_MAP(AssignAdd) = EMPTY_ATTR_MAP;
96 OUTPUT_MAP(AssignAdd) = {{0, OUTPUT_DESC(ref)}};
97 REG_ADPT_DESC(AssignAdd, kNameAssignAdd, ADPT_DESC(AssignAdd))
|
| /third_party/mindspore/mindspore/boost/ |
| D | grad_accumulation.py | 33 return P.AssignAdd()(cumulative_grad, grad / accumulation_step)
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_math_ops.py | 348 class AssignAdd(nn.Cell): class
351 self.op = P.AssignAdd()
470 'block': (P.AssignAdd(), {'exception': ValueError}),
|
| /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/cpp/python_input/gtest_input/pre_activate/ |
| D | insert_tensor_move_for_hccl_op.py | 26 assign_add = P.AssignAdd()
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | model_v0.fbs | 235 AssignAdd,
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_func_declare.h | 273 FUNC_MSOP2SCHEMAOP_DECLARE(AssignAdd)
|
| D | ops_def.cc | 34 OP_TYPE(AssignAdd)
298 OP_SCHEMA_DEF(AssignAdd)
299 OP_SCHEMA_DEF_END(AssignAdd)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 49 from .math_ops import (Abs, ACos, Asin, Asinh, AddN, AccumulateNV2, AssignAdd, AssignSub, Atan2, Ba…
|