Searched refs:output_tf_md (Results 1 – 3 of 3) sorted by relevance
150 auto output_tf_md = mkl_shape_input.GetTfLayout(); in Compute() local152 memory::primitive_desc(output_tf_md, cpu_engine); in Compute()180 auto output_tf_md = MklDnnData<T>::CreateBlockedMemDesc( in Compute() local183 memory::primitive_desc(output_tf_md, cpu_engine); in Compute()
92 auto output_tf_md = input_shape.GetTfLayout(); in ConvertMklToTf() local93 auto output_tf_pd = memory::primitive_desc(output_tf_md, cpu_engine); in ConvertMklToTf()
850 auto output_tf_md = mkl_shape.GetTfLayout(); in ConvertMklToTF() local851 auto output_tf_pd = memory::primitive_desc(output_tf_md, cpu_engine); in ConvertMklToTF()