Searched defs:ksize_h (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | avgpool_fusion.cc | 39 bool CheckAttrValue(const AnfNodePtr &avgpool, int64_t input_c, int64_t output_w, int64_t ksize_h, … in CheckAttrValue()
70 … int64_t *input_c, int64_t *ksize_h, int64_t *ksize_w, int64_t *stride_h, int64_t *stride_w, in GetAttrValue()
106 bool CheckOptimization(int64_t ksize_h, int64_t ksize_w, int64_t stride_h, int64_t stride_w, size_t… in CheckOptimization()
263 int64_t ksize_h = 0; in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | maxpool_with_argmax_v2_impl.cu | 73 … const int ksize_h, const int ksize_w, const int strides_h, const int strides_w, in CalMaxPoolWithArgmaxV2()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | fractional_max_pool_with_fixed_ksize.cc | 116 int64_t ksize_h = 0; in FractionalMaxPoolWithFixedKsizeInferShape() local
|