Searched refs:tensor_length (Results 1 – 3 of 3) sorted by relevance
158 const int tensor_length, const int num_ranks) = 0;460 void InitExpected(std::vector<float>* expected, const int tensor_length, in InitExpected() argument462 expected->resize(tensor_length); in InitExpected()463 for (int i = 0; i < tensor_length; ++i) { in InitExpected()495 void InitExpected(std::vector<float>* expected, const int tensor_length, in InitExpected() argument497 expected->resize(tensor_length); in InitExpected()498 for (int i = 0; i < tensor_length; ++i) { in InitExpected()529 void InitExpected(std::vector<float>* expected, const int tensor_length, in InitExpected() argument531 expected->resize(tensor_length * num_ranks, -1); in InitExpected()533 for (int j = 0; j < tensor_length; ++j, ++i) { in InitExpected()
106 const INDEX_TYPE tensor_length = row_split.size(); in GetMaxWidthRowSplit() local107 if (tensor_length == 0 || tensor_length == 1) { in GetMaxWidthRowSplit()111 for (INDEX_TYPE i = 0; i < tensor_length - 1; ++i) { in GetMaxWidthRowSplit()
188 const int tensor_length = tensor_shape.FlatSize(); in GetMaxWidthRowSplit() local189 if (tensor_length == 0 || tensor_length == 1) { in GetMaxWidthRowSplit()205 for (int i = 1; i < tensor_length; ++i) { in GetMaxWidthRowSplit()