Home
last modified time | relevance | path

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

/ohos5.0/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 __anond68e77050102() variable
172 EXPECT_NE(nullptr, dimsValue);
173 dimsTensor->SetBuffer(dimsValue, sizeof(float));
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dneural_network_runtime.cpp323 int32_t* dimsValue = reinterpret_cast<int32_t*>(data); in ParseInputDimsFromExtensions() local
349 inputDim.emplace_back(dimsValue[dataIndex]); in ParseInputDimsFromExtensions()
350 if (dimsValue[dataIndex] == -1) { in ParseInputDimsFromExtensions()