Home
last modified time | relevance | path

Searched refs:output_tf_pd (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_tfconv_op.h93 auto output_tf_pd = memory::primitive_desc(output_tf_md, cpu_engine); in ConvertMklToTf() local
105 if (input.IsReorderNeeded(output_tf_pd)) { in ConvertMklToTf()
107 CHECK_EQ(input.CheckReorderToOpMem(output_tf_pd, output_tensor), true); in ConvertMklToTf()
Dmkl_reshape_op.cc151 auto output_tf_pd = in Compute() local
166 if (dnn_data_input.CheckReorderToOpMem(output_tf_pd, output_tensor)) { in Compute()
182 auto output_tf_pd = in Compute() local
188 mkl_shape_output.SetMklLayout(&output_tf_pd); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h851 auto output_tf_pd = memory::primitive_desc(output_tf_md, cpu_engine); in ConvertMklToTF() local
855 if (input.IsReorderNeeded(output_tf_pd)) { in ConvertMklToTF()
857 CHECK_EQ(input.CheckReorderToOpMem(output_tf_pd, &output_tensor, &net), in ConvertMklToTF()