Searched defs:pooling_ratio (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | fractional_max_pool.cc | 59 auto pooling_ratio = GetValue<std::vector<float>>(primitive->GetAttr(kPoolingRatio)); in FractionalMaxPoolInferShape() local
119 void FractionalMaxPool::Init(const std::vector<float> pooling_ratio, const bool pseudo_random, cons… in Init()
129 void FractionalMaxPool::set_pooling_ratio(const std::vector<float> pooling_ratio) { in set_pooling_ratio()
|
| D | fractional_avg_pool.cc | 54 auto pooling_ratio = GetValue<std::vector<float>>(primitive->GetAttr(kPoolingRatio)); in FractionalAvgPoolInferShape() local
130 void FractionalAvgPool::Init(const std::vector<float> pooling_ratio, const bool pseudo_random, cons… in Init()
140 void FractionalAvgPool::set_pooling_ratio(const std::vector<float> pooling_ratio) { in set_pooling_ratio()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | fractional_avg_pool_proto.cc | 31 std::vector<float> pooling_ratio; in IMPLEMT_INFERFUNC() local
|
| D | nn_pooling_ops_proto.cc | 511 std::vector<float> pooling_ratio; in IMPLEMT_INFERFUNC() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | fractional_max_pool.cc | 53 AttrValue *pooling_ratio = ctx.GetAttr("pooling_ratio"); in FractionalMaxPoolParamCheck() local
132 std::vector<float> pooling_ratio = ctx.GetAttr("pooling_ratio")->GetListFloat(); in DoCompute() local
|
| D | fractional_avg_pool.cc | 54 AttrValue *pooling_ratio = ctx.GetAttr("pooling_ratio"); in FractionalAvgPoolParamCheck() local
133 std::vector<float> pooling_ratio = ctx.GetAttr("pooling_ratio")->GetListFloat(); in DoCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/ |
| D | fractionalpool_helper.h | 56 std::vector<float> pooling_ratio; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | nn_ops.py | 8322 …def __init__(self, pooling_ratio, pseudo_random=False, overlapping=False, deterministic=False, see… argument
8503 …def __init__(self, pooling_ratio, pseudo_random=False, overlapping=False, deterministic=False, see… argument
|