Searched defs:pool_h (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/vm_impl/ |
| D | vm_me.py | 21 def avg_pooling(x, pool_h, pool_w, stride): argument
48 def avg_pool_grad(dout, origin_shape, pool_h, pool_w, stride): argument
422 def max_pooling(x, pool_h, pool_w, stride): argument
438 def max_pool_grad(x, dout, pool_h, pool_w, stride): argument
453 def max_pool_grad_with_argmax(x, dout, arg_max, pool_h, pool_w, stride): argument
465 def max_pool_with_argmax(x, pool_h, pool_w, stride): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | fractional_max_pool_with_fixed_ksize.cc | 77 int64_t pool_h = ksize_values[0]; in Compute() local
156 …const int output_h, const int output_w, const int pool_h, const int pool_w, CpuKernelContext &ctx)… in FractionalMaxPoolWithFixedKsizeCompute()
208 const int pool_h, const int pool_w) { in ComputeSingleBatch()
|