Searched defs:pad_num (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | unique_with_pad_impl.cu | 31 cudaStream_t cuda_stream, T *pad_num) { in CalUniqueWithPad()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_unique_with_pad.py | 29 def __init__(self, pad_num): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_unique_with_pad_op.py | 29 def __init__(self, pad_num): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_unique_with_pad_op.py | 30 def __init__(self, pad_num): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | unique_with_pad_cpu_kernel.cc | 83 T pad_num = *pad_num_p; in PadOutput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | unique_with_pad.cc | 91 … auto pad_num = std::accumulate(pad_shape.begin(), pad_shape.end(), 1, std::multiplies<int64_t>()); in UniqueWithPadInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | neighbor_exchange_v2_fusion.cc | 977 int64_t pad_num = 1; in CreateSplitGradNodes() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | neighbor_exchange_v2_unify_mindir.cc | 918 int64_t pad_num = 1; in CreateSplitGradNodes() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | array_func.py | 1278 def unique_with_pad(x, pad_num): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/parse/ |
| D | standard_method.py | 1326 def unique_with_pad(x, pad_num): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | tensor.py | 3719 def unique_with_pad(self, pad_num): argument
|