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