Searched refs:SetMklLayout (Results 1 – 15 of 15) sorted by relevance
132 mkl_output_mkl_shape.SetMklLayout(&input1_md); in Compute()230 mkl_output_mkl_shape.SetMklLayout(&output_mkl_md); in Compute()
146 output_mkl_shape.SetMklLayout(&dst_pd); in Compute()
188 mkl_shape_output.SetMklLayout(&output_tf_pd); in Compute()
426 diff_src_mkl_shape.SetMklLayout(&diff_src_pd); in Compute()546 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
220 output_mkl_shape.SetMklLayout(&output_pd); in Compute()
127 mkl_out_shape.SetMklLayout(mkl_context.prim_pooling_fwd, dnnResourceDst); in Compute()317 mkl_out_shape.SetMklLayout(mkl_context.prim_pooling_bwd, in Compute()
206 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()456 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
628 mkl_output_mkl_shape.SetMklLayout(mkl_context.prim_fwd, dnnResourceDst); in Compute()645 mkl_filter_output_mkl_shape.SetMklLayout(mkl_context.prim_fwd, in Compute()1266 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()1338 filter_mkl_shape.SetMklLayout(&filter_pd); in AllocateFilterOutputTensor()1778 summand_mkl_shape.SetMklLayout(&dst_md); in AllocateOutputTensor()
650 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()702 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
108 mkl_out_shape.SetMklLayout(mkl_context.prim_pooling_fwd, dnnResourceDst); in Compute()280 mkl_output_shape.SetMklLayout(mkl_context.prim_pooling_bwd, in Compute()
460 output_mkl_shape->SetMklLayout(output_pd); in AllocateOutputTensor()
663 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()996 dnn_shape_diff_src.SetMklLayout(&diff_src_pd); in Compute()
510 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()672 dnn_shape_diff_src.SetMklLayout(&diff_src_pd); in Compute()
400 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()
172 void SetMklLayout(dnnLayout_t mklLayout) { mklLayout_ = mklLayout; } in SetMklLayout() function174 void SetMklLayout(const void* primitive, size_t resourceType) { in SetMklLayout() function614 inline void SetMklLayout(memory::primitive_desc* pd) { in SetMklLayout() function619 inline void SetMklLayout(memory::desc* md) { in SetMklLayout() function