Browse Source

!9372 add eltwise mkldnn operator scalar support

From: @zhao_ting_v
Reviewed-by: @wuxuejian,@kisnwang
Signed-off-by: @wuxuejian
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
b5fbe86a56
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/eltwise_cpu_kernel.cc

+ 3
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/eltwise_cpu_kernel.cc View File

@@ -51,6 +51,9 @@ dnnl::eltwise_forward::desc EltWiseCPUKernel::GetForwardEltwiseDesc(const CNodeP
void EltWiseCPUKernel::InitKernel(const CNodePtr &kernel_node) { void EltWiseCPUKernel::InitKernel(const CNodePtr &kernel_node) {
MS_EXCEPTION_IF_NULL(kernel_node); MS_EXCEPTION_IF_NULL(kernel_node);
std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0);
if (src_shape.size() == 0) {
src_shape.insert(src_shape.begin(), 1);
}
dnnl::memory::desc src_desc = GetDefaultMemDesc(src_shape); dnnl::memory::desc src_desc = GetDefaultMemDesc(src_shape);


auto desc = GetForwardEltwiseDesc(kernel_node, src_desc); auto desc = GetForwardEltwiseDesc(kernel_node, src_desc);


Loading…
Cancel
Save