Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_input_conversion_op.cc119 mkl_output_mkl_shape.SetMklLayout(&input1_md); in Compute()
220 mkl_output_mkl_shape.SetMklLayout(&output_mkl_md); in Compute()
Dmkl_dequantize_op_test.cc61 mkl_shape.SetMklLayout(&md); in CreateMklInput()
Dmkl_aggregate_ops.cc223 output_mkl_shape.SetMklLayout(&output_pd); in Compute()
Dmkl_conv_grad_input_ops.cc446 diff_src_mkl_shape.SetMklLayout(&diff_src_pd); in Compute()
575 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
Dmkl_softmax_op.cc309 output_mkl_shape.SetMklLayout(&dst_pd); in Compute()
Dmkl_pooling_ops_common.h668 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
720 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
Dmkl_lrn_op.cc202 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
464 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
Dmkl_quantize_op.cc418 output_mkl_shape.SetMklLayout(&dst_md); in Compute()
Dmkl_slice_op.cc494 output_mkl_shape->SetMklLayout(output_pd); in AllocateOutputTensor()
Dmkl_relu_op.cc503 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()
715 dnn_shape_diff_src.SetMklLayout(&diff_src_pd); in Compute()
Dmkl_fused_batch_norm_op.cc911 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()
1340 dnn_shape_diff_src.SetMklLayout(&diff_src_pd); in Compute()
Dmkl_conv_ops.cc1100 output_mkl_shape->SetMklLayout(&dst_md); in AllocateOutputTensor()
1264 filter_mkl_shape.SetMklLayout(&filter_md); in AllocateFilterOutputTensor()
2035 summand_mkl_shape.SetMklLayout(&dst_md); in AllocateOutputTensor()
Dmkl_concat_op.cc737 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()
Dmkl_matmul_ops_common.h472 output_mkl_shape.SetMklLayout(&dst_pd);
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h423 inline void SetMklLayout(memory::desc* md) { in SetMklLayout() function