Home
last modified time | relevance | path

Searched defs:AvgPool (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/st/dyn_shape_dev/
Dtest_avg_pool_grad.py26 class AvgPool(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_avgpool_op.py31 class AvgPool(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_avgpool_op.py31 class AvgPool(nn.Cell): class
/third_party/mindspore/mindspore-src/source/config/
Dsuper_bar_config.json232 "AvgPool": { object
487 "AvgPool": "not support strides > 63.", string
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.h301 AvgPool() : BaseOperator(kNameAvgPool) {} in AvgPool() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/trt_pass/
Dtrt_op_converter.cc351 MS_TRT_CONVERTER_FUNC_REG(AvgPool) { return AddPoolingLayer(node, context, nvinfer1::PoolingType::k… in MS_TRT_CONVERTER_FUNC_REG() argument