Home
last modified time | relevance | path

Searched refs:ScatterDiv (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dmatrix_calculation_ops_declare.cc69 INPUT_MAP(ScatterDiv) = {{1, INPUT_DESC(var)}, {2, INPUT_DESC(indices)}, {3, INPUT_DESC(updates)}};
70 ATTR_MAP(ScatterDiv) = {{"use_locking", ATTR_DESC(use_locking, AnyTraits<bool>())}};
71 OUTPUT_MAP(ScatterDiv) = {{0, OUTPUT_DESC(var)}};
72 REG_ADPT_DESC(ScatterDiv, kNameScatterDiv, ADPT_DESC(ScatterDiv))
Dmatrix_calculation_ops_declare.h50 DECLARE_OP_ADAPTER(ScatterDiv)
51 DECLARE_OP_USE_OUTPUT(ScatterDiv)
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dscatter_arithmetic_cpu_kernel.h44 void ScatterDiv(T *input, const int *indices, const T *updates) const;
124 MS_REG_CPU_KERNEL_T(ScatterDiv,
131 MS_REG_CPU_KERNEL_T(ScatterDiv,
138 MS_REG_CPU_KERNEL_T(ScatterDiv,
Dscatter_arithmetic_cpu_kernel.cc35 {prim::kPrimScatterDiv->name(), &ScatterArithmeticCPUKernel<T>::ScatterDiv}, in InitComputeFunc()
124 void ScatterArithmeticCPUKernel<T>::ScatterDiv(T *input, const int *indices, const T *updates) cons… in ScatterDiv() function in mindspore::kernel::ScatterArithmeticCPUKernel
/third_party/mindspore/tests/ut/python/ops/
Dtest_ops.py530 class ScatterDiv(nn.Cell): class
534 super(ScatterDiv, self).__init__()
535 self.scatter_div = P.ScatterDiv(use_locking)
2802 'block': ScatterDiv((6,), use_locking=True),
2807 'block': ScatterDiv((6,)),
2812 'block': ScatterDiv((3, 4)),
2818 'block': ScatterDiv((6,), np.float16),
2823 'block': ScatterDiv((6,), np.int8),
2828 'block': ScatterDiv((6,), np.uint8),
/third_party/mindspore/mindspore/ops/operations/
D__init__.py28 … SameTypeShape, ScatterAdd, ScatterSub, ScatterMul, ScatterDiv, ScatterMax, ScatterMin,
Darray_ops.py4491 class ScatterDiv(_ScatterOp): class
/third_party/mindspore/tests/st/auto_monad/
Dtest_effect_ops.py159 self.scatter_div = P.ScatterDiv()
/third_party/mindspore/tests/st/ops/cpu/
Dtest_scatter_arithmetic_op.py385 self.scatter_div = P.ScatterDiv(use_locking=lock)
/third_party/mindspore/config/
Dop_info.config485 {"op_name": "ScatterDiv", "inputs": [{"index": 0, "name": "var", "need_compile": false, "param_type…