Home
last modified time | relevance | path

Searched refs:srcs_pd (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_aggregate_ops.cc192 std::vector<memory::primitive_desc> srcs_pd; in Compute() local
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()
197 auto sum_pd = sum::primitive_desc(dst.GetUsrMemDesc(), coeff, srcs_pd); in Compute()
207 src1.CheckReorderToOpMem(srcs_pd[0]); in Compute()
208 src2.CheckReorderToOpMem(srcs_pd[1]); in Compute()
Dmkl_concat_op.cc289 std::vector<memory::primitive_desc> srcs_pd; in Compute() local
308 srcs_pd.push_back(src_mpd); in Compute()
326 srcs_pd.push_back(memory::primitive_desc(src_md, cpu_engine)); in Compute()
343 srcs_pd.push_back(src_mpd); in Compute()
371 srcs[k].CheckReorderToOpMem(srcs_pd[k]); in Compute()
391 auto concat_pd = concat::primitive_desc(concat_dim, srcs_pd); in Compute()