Home
last modified time | relevance | path

Searched refs:KERNEL_SIZE (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dmaxpool_info.cc33 kernel_size_ = GetTupleIntAttr(KERNEL_SIZE); in GetAttrs()
Dconv2d_info.cc64 auto kernel_size_iter = attrs_.find(KERNEL_SIZE); in GetAttrsBase()
66 MS_LOG(ERROR) << name_ << ": Can not find the attribution of " << KERNEL_SIZE; in GetAttrsBase()
667 Attr kernel_size = {KERNEL_SIZE, MakeValue(kernel_size_)}; in CreateConv2DAttrs()
Dops_utils.h214 constexpr char KERNEL_SIZE[] = "kernel_size"; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/
Dpooling_avg_grad_cpu_kernel.cc44 …or<int64_t> kernel_sizes_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, KERNEL_SIZE); in InitKernel()
Dpooling_cpu_kernel.cc44 …or<int64_t> kernel_sizes_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, KERNEL_SIZE); in InitKernel()
Dpooling_max_grad_cpu_kernel.cc42 auto kernel_sizes_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, KERNEL_SIZE); in InitKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcpu_kernel.h42 constexpr char KERNEL_SIZE[] = "kernel_size"; variable