Function test_one_hot

src/opr/test/indexing.cpp:161–164

}

void test_one_hot(int32_t axis, const TensorShapeArray& test_cases) {
    test_one_hot_get(axis, test_cases);
    test_one_hot_set(axis, test_cases);
}

}  // anonymous namespace