Home
last modified time | relevance | path

Searched refs:SetMklLayout (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_input_conversion_op.cc132 mkl_output_mkl_shape.SetMklLayout(&input1_md); in Compute()
230 mkl_output_mkl_shape.SetMklLayout(&output_mkl_md); in Compute()
Dmkl_softmax_op.cc146 output_mkl_shape.SetMklLayout(&dst_pd); in Compute()
Dmkl_reshape_op.cc188 mkl_shape_output.SetMklLayout(&output_tf_pd); in Compute()
Dmkl_conv_grad_input_ops.cc426 diff_src_mkl_shape.SetMklLayout(&diff_src_pd); in Compute()
546 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
Dmkl_aggregate_ops.cc220 output_mkl_shape.SetMklLayout(&output_pd); in Compute()
Dmkl_avgpooling_op.cc127 mkl_out_shape.SetMklLayout(mkl_context.prim_pooling_fwd, dnnResourceDst); in Compute()
317 mkl_out_shape.SetMklLayout(mkl_context.prim_pooling_bwd, in Compute()
Dmkl_lrn_op.cc206 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
456 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
Dmkl_conv_ops.cc628 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()
Dmkl_pooling_ops_common.h650 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
702 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
Dmkl_maxpooling_op.cc108 mkl_out_shape.SetMklLayout(mkl_context.prim_pooling_fwd, dnnResourceDst); in Compute()
280 mkl_output_shape.SetMklLayout(mkl_context.prim_pooling_bwd, in Compute()
Dmkl_slice_op.cc460 output_mkl_shape->SetMklLayout(output_pd); in AllocateOutputTensor()
Dmkl_fused_batch_norm_op.cc663 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()
996 dnn_shape_diff_src.SetMklLayout(&diff_src_pd); in Compute()
Dmkl_relu_op.cc510 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()
672 dnn_shape_diff_src.SetMklLayout(&diff_src_pd); in Compute()
Dmkl_concat_op.cc400 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h172 void SetMklLayout(dnnLayout_t mklLayout) { mklLayout_ = mklLayout; } in SetMklLayout() function
174 void SetMklLayout(const void* primitive, size_t resourceType) { in SetMklLayout() function
614 inline void SetMklLayout(memory::primitive_desc* pd) { in SetMklLayout() function
619 inline void SetMklLayout(memory::desc* md) { in SetMklLayout() function