Get index of MetaData tensor from index 'n' of Data tensor.
| 50 | |
| 51 | // Get index of MetaData tensor from index 'n' of Data tensor. |
| 52 | inline int DataIndexToMetaDataIndex(int n, int total_tensors) { |
| 53 | if (kTensorOrdering == MklTfTensorOrdering::TENSORS_INTERLEAVED) { |
| 54 | // For interleaved ordering, OneDNN tensor follows immediately after |
| 55 | // Tensorflow tensor. |
| 56 | return n + 1; |
| 57 | } else { |
| 58 | CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); |
| 59 | // For contiguous ordering, OneDNN tensor is n+total_tensors / 2 away. |
| 60 | return n + total_tensors / 2; |
| 61 | } |
| 62 | } |
| 63 | |
| 64 | int inline GetTensorDataIndex(int n, int total_tensors) { |
| 65 | if (kTensorOrdering == MklTfTensorOrdering::TENSORS_INTERLEAVED) { |
no outgoing calls
no test coverage detected