Searched refs:input_mkl_md (Results 1 – 3 of 3) sorted by relevance
54 auto input_mkl_md = mkl_shape_input.GetMklLayout(); in SkipReorder() local55 if (mkl_shape_input.GetTfDataFormat() == input_mkl_md.data.format && in SkipReorder()146 auto input_mkl_md = mkl_shape_input.GetMklLayout(); in Compute() local148 dnn_data_input.SetUsrMem(input_mkl_md, &input_tensor); in Compute()
89 auto input_mkl_md = input_shape.GetMklLayout(); in ConvertMklToTf() local95 input.SetUsrMem(input_mkl_md, &input_tensor); in ConvertMklToTf()
849 auto input_mkl_md = mkl_shape.GetMklLayout(); in ConvertMklToTF() local852 input.SetUsrMem(input_mkl_md, &mkl_tensor); in ConvertMklToTF()