Function TEST_F

dnn/test/naive/mesh_indexing.cpp:10–42

using namespace test;

TEST_F(NAIVE, MESH_INDEXING) {
    SmallVector<size_t> init_axes;

    auto multi_axis_index_impl = [this, &init_axes](const TensorNDArray& tensors) {
        auto opr = handle()->create_operator<IndexingMultiAxisVec>();
        OprProxy<IndexingMultiAxisVec> proxy(init_axes);
        proxy.exec(opr.get(), tensors);
    };

    Checker<MeshIndexing> checker(handle());
    checker.set_extra_opr_impl(multi_axis_index_impl);
    size_t idx_size0, idx_size1;
    IndexRNG rng0{idx_size0, 2}, rng1{idx_size1, 3};
    checker.set_dtype(0, dtype::Float32())
            .set_dtype(1, dtype::Float32())
            .set_dtype(2, dtype::Int32())
            .set_dtype(3, dtype::Int32())
            .set_rng(2, &rng0)
            .set_rng(3, &rng1);

    idx_size0 = 23;
    init_axes = {0};
    checker.set_proxy({init_axes})
            .execs({{23}, {100}, {100}})
            .execs({{23, 5}, {100, 5}, {100}});

    idx_size0 = 3;
    init_axes = {1};
    checker.set_proxy(init_axes)
            .execs({{2, 3}, {2, 10}, {10}})
            .execs({{2, 3, 5}, {2, 50, 5}, {50}})
            .execs({{2, 3, 5, 7}, {2, 55, 5, 7}, {55}});
}

TEST_F(NAIVE, BATCHED_MESH_INDEXING) {
    SmallVector<size_t> init_axes;