Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_tfconv_op.h95 auto output_tf_md = input_shape.GetTfLayout(); in ConvertMklToTf() local
107 if (input.IsReorderNeeded(output_tf_md)) { in ConvertMklToTf()
111 input.CheckReorderToOpMem(output_tf_md, output_tensor, context), in ConvertMklToTf()
Dmkl_reshape_op.cc157 auto output_tf_md = mkl_shape_input.GetTfLayout(); in Compute() local
171 if (dnn_data_input.CheckReorderToOpMem(output_tf_md, output_tensor, in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_types.h102 #define OUTPUT_TF_MD output_tf_md
Dmkl_util.h644 auto output_tf_md = input_mkl_shape.GetTfLayout(); in ConvertMklToTF() local
647 if (input.IsReorderNeeded(output_tf_md)) { in ConvertMklToTF()
650 bool status = input.CheckReorderToOpMem(output_tf_md, output_tf_tensor, in ConvertMklToTF()