Home
last modified time | relevance | path

Searched refs:GetTfLayout (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_tfconv_op.h92 auto output_tf_md = input_shape.GetTfLayout(); in ConvertMklToTf()
Dmkl_reshape_op.cc150 auto output_tf_md = mkl_shape_input.GetTfLayout(); in Compute()
Dmkl_input_conversion_op.cc240 auto input_tf_md = mkl_output_mkl_shape.GetTfLayout(); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h237 const dnnLayout_t GetTfLayout() const { return tfLayout_; } in GetTfLayout() function
647 inline const memory::desc GetTfLayout() const { in GetTfLayout() function
663 return IsMklTensor() ? GetMklLayout() : GetTfLayout(); in GetCurLayout()
816 dnnLayout_t output_layout = static_cast<dnnLayout_t>(mkl_shape.GetTfLayout()); in ConvertMklToTF()
850 auto output_tf_md = mkl_shape.GetTfLayout(); in ConvertMklToTF()