Searched refs:UniqueWithPad (Results 1 – 5 of 5) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
D | unique_with_pad_cpu_kernel.h | 41 MS_REG_CPU_KERNEL(UniqueWithPad, 49 MS_REG_CPU_KERNEL(UniqueWithPad, 57 MS_REG_CPU_KERNEL(UniqueWithPad,
|
/third_party/mindspore/tests/st/ops/ascend/ |
D | test_unique_with_pad.py | 30 self.unique_with_pad = P.UniqueWithPad()
|
/third_party/mindspore/mindspore/ops/operations/ |
D | __init__.py | 31 … UniqueWithPad, ScatterNdAdd, ScatterNdSub, ScatterNonAliasingAdd, ReverseV2, Rint,
|
D | array_ops.py | 962 class UniqueWithPad(PrimitiveWithInfer): class
|
/third_party/mindspore/config/ |
D | op_info.config | 11 {"op_name": "UniqueWithPad", "inputs": [{"index": 0, "name": "x", "param_type": "required"}, {"inde…
|