Home
last modified time | relevance | path

Searched defs:MaxPool (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dmax_pool.h37 MaxPool() : BaseOperator(kNameMaxPool) { InitIOName({"x"}, {"output"}); } in MaxPool() function
38 …explicit MaxPool(const std::string k_name) : BaseOperator(k_name) { InitIOName({"x"}, {"output"});… in MaxPool() function
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_maxpool_grad_op.py29 class MaxPool(nn.Cell): class
Dtest_dynamic_ops.py369 class MaxPool(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_maxpool_grad_op.py29 class MaxPool(nn.Cell): class
Dtest_maxpool_op.py31 class MaxPool(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/pi_jit/share/ops/primitive/
Dmaxpool_ops.py14 class MaxPool(Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/
Dtest_dynamic_asr.py313 class MaxPool(nn.Cell): class
/third_party/mindspore/mindspore-src/source/config/
Dsuper_bar_config.json147 "MaxPool": { object
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/trt_pass/
Dtrt_op_converter.cc352 MS_TRT_CONVERTER_FUNC_REG(MaxPool) { return AddPoolingLayer(node, context, nvinfer1::PoolingType::k… in MS_TRT_CONVERTER_FUNC_REG() argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Dnn_ops.py1252 class MaxPool(_Pool): class