Home
last modified time | relevance | path

Searched defs:MaxPoolWithArgmaxV2 (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dmax_pool_with_argmax_v2.h38 …MaxPoolWithArgmaxV2() : BaseOperator(kNameMaxPoolWithArgmaxV2) { InitIOName({"x"}, {"output", "arg… in MaxPoolWithArgmaxV2() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dmaxpool_with_argmax_v2_impl.cu31 __global__ void MaxPoolWithArgmaxV2(const T *input, T *output, S *index, const int inputN, const in… in MaxPoolWithArgmaxV2() function
/third_party/mindspore/mindspore-src/source/config/
Dsuper_bar_config.json68 "MaxPoolWithArgmaxV2": { object
131 "MaxPoolWithArgmaxV2": { object
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Dnn_ops.py9373 class MaxPoolWithArgmaxV2(Primitive): class