Home
last modified time | relevance | path

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

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dargmax_test.cpp69 bool* keepdimsValue = new (std::nothrow) bool(false); in SetArgmaxKeepdims() local
70 EXPECT_NE(nullptr, keepdimsValue); in SetArgmaxKeepdims()
71 tensor->SetBuffer(keepdimsValue, sizeof(bool)); in SetArgmaxKeepdims()
266 int64_t* keepdimsValue = new (std::nothrow) int64_t(0); in HWTEST_F() local
267 EXPECT_NE(nullptr, keepdimsValue); in HWTEST_F()
269 tensor->SetBuffer(keepdimsValue, sizeof(int64_t)); in HWTEST_F()

Completed in 2 milliseconds