Searched refs:ArgMinWithValue (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | argmaxandminwithvalue_gpu_kernel.cc | 35 ArgMinWithValue,
39 ArgMinWithValue,
43 ArgMinWithValue,
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_argminwithvalue_op.py | 30 self.argmin1 = P.ArgMinWithValue(axis1)
31 self.argmin2 = P.ArgMinWithValue(axis2)
32 self.argmin3 = P.ArgMinWithValue()
41 self.argmin = P.ArgMinWithValue(axis)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | argmin_with_value_cpu_kernel.h | 48 MS_REG_CPU_KERNEL_T(ArgMinWithValue, KernelAttr(), ArgMinWithValueCPUKernel, float);
49 MS_REG_CPU_KERNEL_T(ArgMinWithValue, KernelAttr(), ArgMinWithValueCPUKernel, float16);
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | elewise_calculation_ops_declare.h | 69 DECLARE_OP_ADAPTER(ArgMinWithValue)
70 DECLARE_OP_USE_OUTPUT(ArgMinWithValue)
|
| D | elewise_calculation_ops_declare.cc | 320 INPUT_MAP(ArgMinWithValue) = {{1, INPUT_DESC(x)}};
321 ATTR_MAP(ArgMinWithValue) = {{"axis", ATTR_DESC(dimension, AnyTraits<int64_t>())},
323 OUTPUT_MAP(ArgMinWithValue) = {{0, OUTPUT_DESC(indice)}, {1, OUTPUT_DESC(values)}};
324 REG_ADPT_DESC(ArgMinWithValue, kNameArgMinWithValue, ADPT_DESC(ArgMinWithValue))
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_argminwithvalue_op.py | 30 self.argmin = P.ArgMinWithValue(axis=axis, keep_dims=keep_dims)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 27 Rank, Reshape, ResizeNearestNeighbor, ArgMinWithValue, Meshgrid,
|
| D | array_ops.py | 1816 class ArgMinWithValue(PrimitiveWithInfer): class
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_reduce_method_info.py | 360 self.arg_min_with_value = P.ArgMinWithValue(keep_dims=False, axis=-1).shard(strategy2)
447 self.arg_min_with_value = P.ArgMinWithValue(keep_dims=True, axis=-1).shard(strategy2)
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_math_ops.py | 766 @bprop_getters.register(P.ArgMinWithValue)
771 op = P.ArgMinWithValue(axis)
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops.py | 2340 'block': P.ArgMinWithValue(),
|
| /third_party/mindspore/ |
| D | RELEASE.md | 3035 - Fix ArgMinWithValue operator bugs ([!338](https://gitee.com/mindspore/mindspore/pulls/338)).
|
| /third_party/mindspore/config/ |
| D | op_info.config | 75 {"op_name": "ArgMinWithValue", "inputs": [{"index": 0, "name": "x", "param_type": "required"}], "ou…
351 {"op_name": "ArgMinWithValue", "inputs": [{"index": 0, "name": "x", "need_compile": false, "param_t…
|