Searched refs:input_mkl_tensors (Results 1 – 1 of 1) sorted by relevance
914 OpInputList input_mkl_tensors; in GetMklShapeList() local915 GetMklInputList(ctext, strings::StrCat("mkl_", name), &input_mkl_tensors); in GetMklShapeList()917 for (int i = 0; i < input_mkl_tensors.size(); i++) { in GetMklShapeList()919 input_mkl_tensors[i].flat<uint8>().data(), in GetMklShapeList()920 input_mkl_tensors[i].flat<uint8>().size() * sizeof(uint8)); in GetMklShapeList()928 OpInputList input_mkl_tensors; in GetMklShapeList() local929 GetMklInputList(ctext, strings::StrCat("mkl_", name), &input_mkl_tensors); in GetMklShapeList()931 for (int i = 0; i < input_mkl_tensors.size(); i++) { in GetMklShapeList()933 input_mkl_tensors[i].flat<uint8>().data(), in GetMklShapeList()934 input_mkl_tensors[i].flat<uint8>().size() * sizeof(uint8)); in GetMklShapeList()