| 712 | } |
| 713 | |
| 714 | inline void GetMklShapeList(OpKernelContext* ctext, StringPiece name, |
| 715 | MklDnnShapeList* mkl_shapes) { |
| 716 | OpInputList input_mkl_tensors; |
| 717 | GetMklInputList(ctext, strings::StrCat("mkl_", name), &input_mkl_tensors); |
| 718 | |
| 719 | for (int i = 0; i < input_mkl_tensors.size(); i++) { |
| 720 | (*mkl_shapes)[i].DeSerializeMklDnnShape( |
| 721 | input_mkl_tensors[i].flat<uint8>().data(), |
| 722 | input_mkl_tensors[i].flat<uint8>().size() * sizeof(uint8)); |
| 723 | } |
| 724 | } |
| 725 | |
| 726 | /// Get shape of input tensor pointed by 'input_idx' in TensorShape format. |
| 727 | /// If the input tensor is in OneDNN layout, then obtains TensorShape from |
no test coverage detected