Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dpow_builder_test.cpp55 std::shared_ptr<NNTensor> shiftTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveShift() local
58 shiftTensor->SetBuffer(shiftValue, sizeof(float)); in SaveShift()
59 m_allTensors.emplace_back(shiftTensor); in SaveShift()
178 std::shared_ptr<NNTensor> shiftTensor = TransToNNTensor(OH_NN_INT64, m_shiftDim, variable
181 shiftTensor->SetBuffer(shiftValue, sizeof(shiftValue));
182 m_allTensors.emplace_back(shiftTensor);
252 std::shared_ptr<NNTensor> shiftTensor = TransToNNTensor(OH_NN_FLOAT32, m_shiftDim, variable
254 m_allTensors.emplace_back(shiftTensor);
H A Dexp_test.cpp76 std::shared_ptr<NNTensor> shiftTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveShift() local
79 shiftTensor->SetBuffer(shiftValue, sizeof(float)); in SaveShift()
80 m_allTensors.emplace_back(shiftTensor); in SaveShift()
241 std::shared_ptr<NNTensor> shiftTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
245 shiftTensor->SetBuffer(shiftValue, sizeof(int64_t));
246 m_allTensors.emplace_back(shiftTensor);
359 std::shared_ptr<NNTensor> shiftTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
361 m_allTensors.emplace_back(shiftTensor);