Home
last modified time | relevance | path

Searched refs:shapeTensor (Results 1 – 3 of 3) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dbroadcast_to_test.cpp53 std::shared_ptr<NNTensor> shapeTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
57 shapeTensor->SetBuffer(shapeValue, sizeof(int64_t) * shapeSize); in SaveParamsTensor()
58 m_allTensors.emplace_back(shapeTensor); in SaveParamsTensor()
163 std::shared_ptr<NNTensor> shapeTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
167 shapeTensor->SetBuffer(shapeValue, sizeof(float) * shapeSize);
168 m_allTensors.emplace_back(shapeTensor);
199 std::shared_ptr<NNTensor> shapeTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
201 m_allTensors.emplace_back(shapeTensor);
H A Dcrop_test.cpp62 std::shared_ptr<NNTensor> shapeTensor; in SetInputAndShape() local
63 shapeTensor = TransToNNTensor(OH_NN_FLOAT32, m_shapeDim, nullptr, OH_NN_TENSOR); in SetInputAndShape()
64 m_allTensors.emplace_back(shapeTensor); in SetInputAndShape()
H A Dl2_normalize_test.cpp341 std::shared_ptr<NNTensor> shapeTensor = TransToNNTensor(OH_NN_INT8, m_paramDim, variable
343 m_allTensors.emplace_back(shapeTensor);