Home
last modified time | relevance | path

Searched refs:axisValueTest (Results 1 – 2 of 2) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dfullconnection_with_axis_test.cpp216 int32_t *axisValueTest = new (std::nothrow) int32_t(0); variable
217 EXPECT_NE(nullptr, axisValueTest);
218 tensor->SetBuffer(axisValueTest, sizeof(int32_t));
318 int64_t *axisValueTest = new (std::nothrow) int64_t[2]{0, 0}; variable
319 EXPECT_NE(nullptr, axisValueTest);
320 tensor->SetBuffer(axisValueTest, 2 * sizeof(int64_t));
H A Dargmax_test.cpp241 float* axisValueTest = new (std::nothrow) float(0); variable
242 EXPECT_NE(nullptr, axisValueTest);
244 tensor->SetBuffer(axisValueTest, sizeof(float));