Searched refs:common_md (Results 1 – 2 of 2) sorted by relevance
181 memory::desc common_md({}, memory::data_undef, memory::format_undef); in Compute() local183 common_md = input1_in_mkl_format ? md1 : md2; in Compute()184 dst.SetUsrMem(common_md); in Compute()188 common_md = md1; in Compute()189 dst.SetUsrMem(common_md); in Compute()194 srcs_pd.push_back(memory::primitive_desc(common_md, cpu_engine)); in Compute()196 srcs_pd.push_back(memory::primitive_desc(common_md, cpu_engine)); in Compute()
227 memory::desc common_md; member in tensorflow::MklEltwiseBwdParams233 const memory::desc& common_md, algorithm alg_kind, in MklEltwiseBwdParams() argument236 common_md(common_md), in MklEltwiseBwdParams()248 static_cast<mkldnn::memory::format>(bwdParams.common_md.data.format); in MklEltwiseBwdPrimitive()250 static_cast<mkldnn::memory::format>(bwdParams.common_md.data.format); in MklEltwiseBwdPrimitive()304 std::shared_ptr<memory::desc> common_md; member329 common_md(nullptr), in EltwiseBwdContext()342 context_.src_md.reset(new memory::desc(bwdParams.common_md.data)); in Setup()343 context_.diff_dst_md.reset(new memory::desc(bwdParams.common_md.data)); in Setup()392 static_cast<mkldnn::memory::format>(bwdParams.common_md.data.format); in Get()[all …]