Searched refs:total_tensors (Results 1 – 1 of 1) sorted by relevance
47 inline int DataIndexToMetaDataIndex(int n, int total_tensors) { in DataIndexToMetaDataIndex() argument55 return n + total_tensors / 2; in DataIndexToMetaDataIndex()59 int inline GetTensorDataIndex(int n, int total_tensors) { in GetTensorDataIndex() argument68 int inline GetTensorMetaDataIndex(int n, int total_tensors) { in GetTensorMetaDataIndex() argument71 int tidx = GetTensorDataIndex(n, total_tensors); in GetTensorMetaDataIndex()72 return DataIndexToMetaDataIndex(tidx, total_tensors); in GetTensorMetaDataIndex()