Home
last modified time | relevance | path

Searched refs:topKValue (Results 1 - 1 of 1) sorted by relevance

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dargmax_test.cpp79 int64_t* topKValue = new (std::nothrow) int64_t(0); in SetArgmaxTopK() local
80 EXPECT_NE(nullptr, topKValue); in SetArgmaxTopK()
81 tensor->SetBuffer(topKValue, sizeof(int64_t)); in SetArgmaxTopK()

Completed in 2 milliseconds