Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dlrn_test.cpp71 std::shared_ptr<NNTensor> alphaTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAlpha() local
74 alphaTensor->SetBuffer(alphaValue, sizeof(float)); in SaveAlpha()
75 m_allTensors.emplace_back(alphaTensor); in SaveAlpha()
251 std::shared_ptr<NNTensor> alphaTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
254 alphaTensor->SetBuffer(alphaValue, sizeof(int64_t));
255 m_allTensors.emplace_back(alphaTensor);
464 std::shared_ptr<NNTensor> alphaTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
466 m_allTensors.emplace_back(alphaTensor);