Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_aggregate_ops.cc136 md1 = src1_mkl_shape.GetMklLayout(); in Compute()
165 md2 = src2_mkl_shape.GetMklLayout(); in Compute()
169 md1 = src1_mkl_shape.GetMklLayout(); in Compute()
170 md2 = src2_mkl_shape.GetMklLayout(); in Compute()
Dmkl_input_conversion_op.cc106 auto input0_md = input_shape_0.GetMklLayout(); in Compute()
107 auto input1_md = input_shape_1.GetMklLayout(); in Compute()
229 auto output_mkl_md = mkl_shape->GetMklLayout(); in Compute()
Dmkl_concat_op.cc305 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute()
316 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute()
414 dnn_shape_dst.IsMklTensor() ? dnn_shape_dst.GetMklLayout() : dst_md; in Compute()
509 int fmt = static_cast<int>(input_shapes[k].GetMklLayout().data.format); in FindMklCommonFormat()
517 input_shapes[0].GetMklLayout().data.format); in FindMklCommonFormat()
Dmkl_reshape_op.cc54 auto input_mkl_md = mkl_shape_input.GetMklLayout(); in SkipReorder()
146 auto input_mkl_md = mkl_shape_input.GetMklLayout(); in Compute()
Dmkl_avgpooling_op.cc134 mkl_out_shape.GetMklLayout())) / in Compute()
325 mkl_out_shape.GetMklLayout())) / in Compute()
491 ? dnn_shape_input.GetMklLayout() in Compute()
637 ? grad_mkl_shape.GetMklLayout() in Compute()
Dmkl_tfconv_op.h89 auto input_mkl_md = input_shape.GetMklLayout(); in ConvertMklToTf()
Dmkl_maxpooling_op.cc116 mkl_out_shape.GetMklLayout())) / in Compute()
289 static_cast<dnnLayout_t>(mkl_output_shape.GetMklLayout())) / in Compute()
550 ? dnn_shape_input.GetMklLayout() in Compute()
748 ? grad_mkl_shape.GetMklLayout() in Compute()
Dmkl_relu_op.cc474 src_md = dnn_shape_src.GetMklLayout(); in Compute()
596 src_md = dnn_shape_src.GetMklLayout(); in Compute()
608 diff_dst_md = dnn_shape_diff_dst.GetMklLayout(); in Compute()
623 src_md = dnn_shape_src.GetMklLayout(); in Compute()
624 diff_dst_md = dnn_shape_diff_dst.GetMklLayout(); in Compute()
Dmkl_softmax_op.cc122 ? src_mkl_shape.GetMklLayout() in Compute()
Dmkl_conv_grad_input_ops.cc386 ? filter_mkl_shape.GetMklLayout() in Compute()
397 ? diff_dst_mkl_shape.GetMklLayout() in Compute()
Dmkl_conv_ops.cc637 static_cast<dnnLayout_t>(mkl_output_mkl_shape.GetMklLayout())) / in Compute()
659 mkl_filter_output_mkl_shape.GetMklLayout())) / in Compute()
1031 ? src_mkl_shape.GetMklLayout() in Compute()
1046 ? filter_mkl_shape.GetMklLayout() in Compute()
1771 auto dst_md = summand_mkl_shape.GetMklLayout(); in AllocateOutputTensor()
1823 ? summand_mkl_shape.GetMklLayout() in AllocateOutputTensor()
Dmkl_conv_grad_filter_ops.cc444 ? src_mkl_shape.GetMklLayout() in Compute()
452 ? diff_dst_mkl_shape.GetMklLayout() in Compute()
Dmkl_fused_batch_norm_op.cc616 ? dnn_shape_src.GetMklLayout() in Compute()
937 ? dnn_shape_src.GetMklLayout() in Compute()
941 ? dnn_shape_diff_dst.GetMklLayout() in Compute()
Dmkl_pooling_ops_common.h615 ? input_mkl_shape.GetMklLayout() in ConfigureInput()
722 ? input_gradient_mkl_shape.GetMklLayout() in ConfigureInputGradient()
Dmkl_slice_op.cc400 auto input_md = input_mkl_shape.GetMklLayout(); in ComputeMklSlice()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h236 const dnnLayout_t GetMklLayout() const { return mklLayout_; } in GetMklLayout() function
487 CompareMklDnnLayouts(this->GetMklLayout(),
488 input_shape.GetMklLayout());
624 inline const memory::desc GetMklLayout() const { in GetMklLayout() function
663 return IsMklTensor() ? GetMklLayout() : GetTfLayout(); in GetCurLayout()
849 auto input_mkl_md = mkl_shape.GetMklLayout(); in ConvertMklToTF()