Searched refs:SetMklLayout (Results 1 – 16 of 16) sorted by relevance
131 mkl_output_mkl_shape.SetMklLayout(&input1_md); in Compute()232 mkl_output_mkl_shape.SetMklLayout(&output_mkl_md); in Compute()
70 mkl_shape.SetMklLayout(&md); in CreateMklInput()
110 output_mkl_shape.SetMklLayout(&dst_md); in Compute()
445 diff_src_mkl_shape.SetMklLayout(&diff_src_pd); in Compute()572 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
223 output_mkl_shape.SetMklLayout(&output_pd); in Compute()
286 output_mkl_shape.SetMklLayout(&dst_pd); in Compute()
201 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()463 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
653 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()705 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
409 output_mkl_shape.SetMklLayout(&dst_md); in Compute()
486 output_mkl_shape->SetMklLayout(output_pd); in AllocateOutputTensor()
501 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()712 dnn_shape_diff_src.SetMklLayout(&diff_src_pd); in Compute()
864 output_mkl_shape->SetMklLayout(&dst_md); in AllocateOutputTensor()1019 filter_mkl_shape.SetMklLayout(&filter_md); in AllocateFilterOutputTensor()1762 summand_mkl_shape.SetMklLayout(&dst_md); in AllocateOutputTensor()
850 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()1248 dnn_shape_diff_src.SetMklLayout(&diff_src_pd); in Compute()
399 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
725 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()
425 inline void SetMklLayout(memory::desc* md) { in SetMklLayout() function