Searched refs:normalizeAndCheckIndex (Results 1 – 4 of 4) sorted by relevance
8 int64_t normalizeAndCheckIndex(int64_t idx, int64_t list_size) { in normalizeAndCheckIndex() function334 auto norm_dim = normalizeAndCheckIndex(dim, indices.size()); in computeChunk()547 auto norm_concat_dim = normalizeAndCheckIndex(concat_dim, outputShape.size()); in computeCatWoConditionals()641 auto dim = normalizeAndCheckIndex(dim_, axes.size()); in computeCat()
48 normalizeAndCheckIndex(std::get<int64_t>(inputs[1]), rank); in computeSoftmax()
15 int64_t normalizeAndCheckIndex(int64_t idx, int64_t list_size);
48 int64_t normalizeAndCheckIndex(int64_t idx, int64_t list_size);