Home
last modified time | relevance | path

Searched refs:ScatterMax (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dmatrix_calculation_ops_declare.cc39 INPUT_MAP(ScatterMax) = {{1, INPUT_DESC(var)}, {2, INPUT_DESC(indices)}, {3, INPUT_DESC(updates)}};
40 ATTR_MAP(ScatterMax) = {{"use_locking", ATTR_DESC(use_locking, AnyTraits<bool>())}};
41 OUTPUT_MAP(ScatterMax) = {{0, OUTPUT_DESC(var)}};
42 REG_ADPT_DESC(ScatterMax, kNameScatterMax, ADPT_DESC(ScatterMax))
Dmatrix_calculation_ops_declare.h35 DECLARE_OP_ADAPTER(ScatterMax)
36 DECLARE_OP_USE_OUTPUT(ScatterMax)
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dscatter_arithmetic_cpu_kernel.h45 void ScatterMax(T *input, const int *indices, const T *updates) const;
145 MS_REG_CPU_KERNEL_T(ScatterMax,
152 MS_REG_CPU_KERNEL_T(ScatterMax,
159 MS_REG_CPU_KERNEL_T(ScatterMax,
Dscatter_arithmetic_cpu_kernel.cc36 {prim::kPrimScatterMax->name(), &ScatterArithmeticCPUKernel<T>::ScatterMax}, in InitComputeFunc()
149 void ScatterArithmeticCPUKernel<T>::ScatterMax(T *input, const int *indices, const T *updates) cons… in ScatterMax() function in mindspore::kernel::ScatterArithmeticCPUKernel
/third_party/mindspore/mindspore/ops/operations/
D__init__.py28 … SameTypeShape, ScatterAdd, ScatterSub, ScatterMul, ScatterDiv, ScatterMax, ScatterMin,
Darray_ops.py4062 class ScatterMax(_ScatterOp): class
/third_party/mindspore/tests/ut/python/ops/
Dtest_ops.py426 class ScatterMax(nn.Cell): class
430 super(ScatterMax, self).__init__()
431 self.scatter_max = P.ScatterMax(use_locking)
2655 'block': ScatterMax(use_locking=True),
2660 'block': ScatterMax(),
2665 'block': ScatterMax(),
2670 'block': ScatterMax(np.float16),
2675 'block': ScatterMax(np.int32),
/third_party/mindspore/tests/st/auto_monad/
Dtest_effect_ops.py184 self.scatter_max = P.ScatterMax()
/third_party/mindspore/tests/st/ops/cpu/
Dtest_scatter_arithmetic_op.py451 self.scatter_max = P.ScatterMax(use_locking=lock)
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_array_ops.py794 @bprop_getters.register(P.ScatterMax)
/third_party/mindspore/config/
Dop_info.config481 {"op_name": "ScatterMax", "inputs": [{"index": 0, "name": "var", "need_compile": false, "param_type…