Searched refs:output_pd (Results 1 – 2 of 2) sorted by relevance
219 auto output_pd = sum_pd.dst_desc(); in Compute() local220 dst.SetUsrMem(output_pd); in Compute()223 output_mkl_shape.SetMklLayout(&output_pd); in Compute()228 output_tf_shape.AddDim((output_pd.get_size() / sizeof(T))); in Compute()
442 auto output_pd = output_md; in ComputeMklSlice() local443 AllocateOutputTensor(context, input_mkl_shape, &output_pd, size_dims, in ComputeMklSlice()471 memory::desc* output_pd, in AllocateOutputTensor() argument486 output_mkl_shape->SetMklLayout(output_pd); in AllocateOutputTensor()491 output_tf_shape.AddDim(output_pd->get_size() / sizeof(T)); in AllocateOutputTensor()