Searched defs:pool_h (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/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
|