Searched refs:avg_pooling (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/ut/python/pynative_mode/vm/ |
| D | test_vm.py | 27 out = vm.avg_pooling(input_data, pool_h=2, pool_w=2, stride=1)
41 dout = vm.avg_pooling(input_data, pool_h=2, pool_w=2, stride=1)
|
| /third_party/mindspore/tests/vm_impl/ |
| D | vm_interface.py | 25 setattr(vm, 'avg_pooling', avg_pooling)
|
| D | nn_ops_vm_impl.py | 187 out = vm.avg_pooling(x, self.kernel_size[-2], self.kernel_size[-1], self.strides[-2])
|
| D | vm_me.py | 21 def avg_pooling(x, pool_h, pool_w, stride): function
|