| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_maxpool_grad_grad_with_argmax_gpu_op.py | 42 def test_maxpool_grad_grad_argmax_fp16(argmax_type): argument
69 def test_maxpool_grad_grad_argmax_fp32(argmax_type): argument
|
| D | test_maxpool_with_argmax_v2_gpu_op.py | 29 def __init__(self, kernel_size, strides, pads, dilation, ceil_mode, argmax_type=mstype.int64): argument
|
| D | test_maxpool_grad_with_argmax_v2_gpu_op.py | 31 def __init__(self, kernel_size, strides, pads, dilation, ceil_mode, argmax_type=mstype.int64): argument
|
| D | test_maxpool3d_with_argmax_op.py | 31 data_format="NCDHW", argmax_type=mstype.int64): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_maxpool_grad_grad_with_argmax_cpu_op.py | 43 def test_maxpool_grad_grad_argmax_fp16(argmax_type): argument
74 def test_maxpool_grad_grad_argmax_fp32(argmax_type): argument
|
| D | test_max_pool_with_argmax_v2_cpu_op.py | 30 def __init__(self, kernel_size, strides, pads, dilation, ceil_mode, argmax_type=mstype.int64): argument
|
| D | test_max_pool_grad_with_argmax_v2_cpu_op.py | 31 def __init__(self, kernel_size, strides, pads, dilation, ceil_mode, argmax_type=mstype.int64): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_max_pool_with_indices.cc | 43 ValuePtr argmax_type; member
69 auto argmax_type = param.argmax_type->ToAbstract(); in TEST_P() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_maxpool_with_argmax_v2.py | 26 def __init__(self, kernel_size, strides, pads, dilation, ceil_mode, argmax_type=mstype.int64): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | max_pool_grad_with_argmax_v2.cc | 31 bool ceil_mode, const TypeId &argmax_type) { in Init()
64 void MaxPoolGradWithArgmaxV2::set_argmax_type(const TypeId &argmax_type) { in set_argmax_type()
|
| D | max_unpool3d_grad.cc | 63 auto argmax_type = input_args[kInputIndex2]->GetType(); in MaxUnpool3DGradInferType() local
|
| D | max_unpool2d_grad.cc | 63 auto argmax_type = input_args[kInputIndex2]->GetType(); in MaxUnpool2DGradInferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | fractional_max_pool_3d_grad_with_fixed_ksize.cc | 234 … DataType argmax_type) { in DoComputeWithArgmaxType()
250 auto argmax_type = ctx.Input(2)->GetDataType(); in Compute() local
|
| D | fractional_max_pool_3d_with_fixed_ksize.cc | 329 DataType argmax_type) { in DoComputeWithArgmaxType()
344 auto argmax_type = ctx.Output(1)->GetDataType(); in DoComputeWithRandomSamplesType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | max_pool_with_argmax_v2.cc | 32 const TypeId &argmax_type) { in Init()
64 void MaxPoolWithArgmaxV2::set_argmax_type(const TypeId &argmax_type) { in set_argmax_type()
|
| D | max_pool3d_with_argmax.cc | 52 const Format &format, const TypeId &argmax_type) { in Init()
90 void MaxPool3DWithArgmax::set_argmax_type(const TypeId &argmax_type) { in set_argmax_type()
|
| D | max_unpool2d.cc | 162 auto argmax_type = input_args[kInputIndex1]->GetType(); in MaxUnpool2DInferType() local
|
| D | max_unpool3d.cc | 177 auto argmax_type = input_args[kInputIndex1]->GetType(); in MaxUnpool3DInferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | max_pool_with_mask.cc | 49 … const BoolImmPtr &ceil_mode, const Int64ImmPtr &argmax_type) { in MaxPoolWithMaskAscendCustomize()
|
| D | max_pool_with_indices.cc | 49 … const BoolImmPtr &ceil_mode, const Int64ImmPtr &argmax_type) { in MaxPoolWithIndicesAscendCustomize()
|
| D | max_pool_grad_with_mask.cc | 50 … const BoolImmPtr &ceil_mode, const Int64ImmPtr &argmax_type) { in MaxPoolGradWithMaskAscendCustomize()
|
| D | max_pool_grad_with_indices.cc | 52 const Int64ImmPtr &argmax_type) { in MaxPoolGradWithIndicesAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_ops_maxpool_with_argmax_v2.py | 27 def __init__(self, kernel_size, strides, pads, dilation, ceil_mode, argmax_type=mstype.int64): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | fractional_max_pool3d_grad_with_fixed_ksize_cpu_kernel.cc | 203 …::vector<KernelTensor *> &inputs, const std::vector<KernelTensor *> &outputs, TypeId argmax_type) { in DoComputeWithArgmaxType()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | nn_pooling_ops_proto.cc | 399 std::string argmax_type; in CUST_IMPLEMT_INFERFUNC() local
486 std::string argmax_type; in CUST_IMPLEMT_VERIFIER() local
668 std::string argmax_type; in CUST_IMPLEMT_VERIFIER() local
|