Home
last modified time | relevance | path

Searched defs:AssignAdd (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dassign_add.h34 AssignAdd() : PrimitiveC(kNameAssignAdd) { InitIOName({"ref", "value"}, {"output"}); } in AssignAdd() function
/third_party/mindspore/mindspore/_extends/graph_kernel/expanders/
Dassign_add.py20 class AssignAdd(Expander): class
/third_party/mindspore/tests/st/ops/gpu/
Dtest_assign_add_op.py25 class AssignAdd(nn.Cell): class
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dassign_add_impl.cu21 __global__ void AssignAdd(const size_t size, T* ref, const T* value, T* output) { in AssignAdd() function
/third_party/mindspore/tests/st/ops/cpu/
Dtest_assign_add_op.py25 class AssignAdd(nn.Cell): class
/third_party/mindspore/tests/st/ops/graph_kernel/
Dtest_assign_add.py25 class AssignAdd(nn.Cell): class
/third_party/mindspore/tests/syntax/simple_expression/
Dtest_assignment_ops.py188 class AssignAdd(nn.Cell): class
/third_party/mindspore/tests/ut/python/ops/
Dtest_math_ops.py348 class AssignAdd(nn.Cell): class
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Darithmetic_cpu_kernel.cc65 void ArithmeticCPUKernel<T>::AssignAdd(T *input1, const T *input2, T *out) const { in AssignAdd() function in mindspore::kernel::ArithmeticCPUKernel
/third_party/mindspore/mindspore/ops/operations/
Dmath_ops.py225 class AssignAdd(PrimitiveWithInfer): class