| 62 | } |
| 63 | |
| 64 | int inline GetTensorDataIndex(int n, int total_tensors) { |
| 65 | if (kTensorOrdering == MklTfTensorOrdering::TENSORS_INTERLEAVED) { |
| 66 | return 2 * n; // index corresponding to nth input/output tensor |
| 67 | } else { |
| 68 | CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); |
| 69 | return n; |
| 70 | } |
| 71 | } |
| 72 | |
| 73 | int inline GetTensorMetaDataIndex(int n, int total_tensors) { |
| 74 | // Get index for TensorData first and then use mapping function |
no outgoing calls
no test coverage detected