Home
last modified time | relevance | path

Searched defs:MaxPool3DWithArgmax (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dmax_pool3d_with_argmax.h37 MaxPool3DWithArgmax() : BaseOperator(kNameMaxPool3DWithArgmax) { in MaxPool3DWithArgmax() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dmaxpool3d_with_argmax_impl.cu29 __global__ void MaxPool3DWithArgmax(const T *input, T *output, S *index, const int inputD, const in… in MaxPool3DWithArgmax() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dnn_pooling_ops_proto.cc310 CUST_IMPLEMT_INFERFUNC(MaxPool3DWithArgmax, MaxPool3DWithArgmaxInferShape) { in CUST_IMPLEMT_INFERFUNC() argument
415 CUST_IMPLEMT_VERIFIER(MaxPool3DWithArgmax, MaxPool3DWithArgmaxVerify) { in CUST_IMPLEMT_VERIFIER() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_info_cfg/
Dcust_aicpu_kernel.ini2098 [MaxPool3DWithArgmax] section
2099 opInfo.subTypeOfInferShape=1
2100 opInfo.opsFlag=OPS_FLAG_CLOSE
2101 opInfo.engine=DNN_VM_AICPU
2102 opInfo.flagPartial=False
2103 opInfo.computeCost=100
2104 opInfo.flagAsync=False
2105 opInfo.opKernelLib=CUSTAICPUKernel
2106 opInfo.formatAgnostic=False
2107 opInfo.kernelSo=libcust_cpu_kernels.so
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Dnn_ops.py1918 class MaxPool3DWithArgmax(Primitive): class