Home
last modified time | relevance | path

Searched refs:GetMklLayout (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_input_conversion_op.cc104 auto input0_md = input_shape_0.GetMklLayout(); in Compute()
105 auto input1_md = input_shape_1.GetMklLayout(); in Compute()
231 auto output_mkl_md = mkl_shape->GetMklLayout(); in Compute()
Dmkl_dequantize_op.cc85 ? src_mkl_shape.GetMklLayout() in Compute()
97 dst_md = memory::desc(src_mkl_shape.GetMklLayout().data); in Compute()
Dmkl_avgpooling_op.cc98 ? dnn_shape_input.GetMklLayout() in Compute()
229 ? orig_input_mkl_shape.GetMklLayout() in Compute()
236 ? grad_mkl_shape.GetMklLayout() in Compute()
Dmkl_maxpooling_op.cc108 ? dnn_shape_input.GetMklLayout() in Compute()
286 ? orig_input_mkl_shape.GetMklLayout() in Compute()
293 ? grad_mkl_shape.GetMklLayout() in Compute()
Dmkl_tfconv_op.h92 auto input_mkl_md = input_shape.GetMklLayout(); in ConvertMklToTf()
Dmkl_matmul_op_fused.cc166 ? add_mkl_shape.GetMklLayout() in Compute()
213 ? src_mkl_shape.GetMklLayout() in Compute()
Dmkl_concat_op.cc610 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute()
621 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute()
738 dst_md = dnn_shape_dst.GetMklLayout(); in Compute()
769 dst_md = dnn_shape_dst.IsMklTensor() ? dnn_shape_dst.GetMklLayout() in Compute()
Dmkl_relu_op.cc467 src_md = dnn_shape_src.GetMklLayout(); in Compute()
632 src_md = dnn_shape_src.GetMklLayout(); in Compute()
645 diff_dst_md = dnn_shape_diff_dst.GetMklLayout(); in Compute()
661 src_md = dnn_shape_src.GetMklLayout(); in Compute()
662 diff_dst_md = dnn_shape_diff_dst.GetMklLayout(); in Compute()
Dmkl_reshape_op.cc153 auto input_mkl_md = mkl_shape_input.GetMklLayout(); in Compute()
Dmkl_conv_grad_input_ops.cc401 ? filter_mkl_shape.GetMklLayout() in Compute()
413 ? diff_dst_mkl_shape.GetMklLayout() in Compute()
Dmkl_qmatmul_op.cc211 ? src_mkl_shape.GetMklLayout() in Compute()
218 ? weight_mkl_shape.GetMklLayout() in Compute()
Dmkl_aggregate_ops.cc191 md = src_mkl_shape.GetMklLayout(); in Compute()
Dmkl_conv_grad_filter_ops.cc442 ? src_mkl_shape.GetMklLayout() in Compute()
450 ? diff_dst_mkl_shape.GetMklLayout() in Compute()
Dmkl_conv_ops.cc613 ? src_mkl_shape.GetMklLayout() in Compute()
627 ? filter_mkl_shape.GetMklLayout() in Compute()
904 ? add_mkl_shape.GetMklLayout() in AllocateOutputTensor()
1754 auto dst_md = summand_mkl_shape.GetMklLayout(); in AllocateOutputTensor()
1818 ? summand_mkl_shape.GetMklLayout() in AllocateOutputTensor()
Dmkl_quantize_op.cc372 ? src_mkl_shape.GetMklLayout() in Compute()
Dmkl_fused_batch_norm_op.cc773 ? dnn_shape_src.GetMklLayout() in Compute()
1174 ? dnn_shape_src.GetMklLayout() in Compute()
1178 ? dnn_shape_diff_dst.GetMklLayout() in Compute()
Dmkl_slice_op.cc414 auto input_md = input_mkl_shape.GetMklLayout(); in ComputeMklSlice()
Dmkl_pooling_ops_common.h617 ? input_mkl_shape.GetMklLayout() in ConfigureInput()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h292 const mkldnn_memory_desc_t& cur_md = (this->GetMklLayout()).data;
294 input_shape.GetMklLayout().data;
430 inline const memory::desc GetMklLayout() const { in GetMklLayout() function
483 return IsMklTensor() ? GetMklLayout() : GetTfLayout(); in GetCurLayout()
643 auto input_mkl_md = input_mkl_shape.GetMklLayout(); in ConvertMklToTF()