Home
last modified time | relevance | path

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

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dtile_builder_test.cpp59 int64_t* dimsValue = new (std::nothrow) int64_t[1] {0}; in SaveDimsTensor() local
60 EXPECT_NE(nullptr, dimsValue); in SaveDimsTensor()
61 dimsTensor->SetBuffer(dimsValue, sizeof(int64_t)); in SaveDimsTensor()
171 float* dimsValue = new (std::nothrow) float[1] {0.0f}; in HWTEST_F() local
172 EXPECT_NE(nullptr, dimsValue); in HWTEST_F()
173 dimsTensor->SetBuffer(dimsValue, sizeof(float)); in HWTEST_F()
/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dneural_network_runtime.cpp321 int32_t* dimsValue = reinterpret_cast<int32_t*>(data); in ParseInputDimsFromExtensions() local
347 inputDim.emplace_back(dimsValue[dataIndex]); in ParseInputDimsFromExtensions()
348 if (dimsValue[dataIndex] == -1) { in ParseInputDimsFromExtensions()

Completed in 3 milliseconds