Searched refs:shiftTensor (Results 1 - 2 of 2) sorted by relevance
| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | exp_test.cpp | 76 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,
in HWTEST_F() local
245 shiftTensor->SetBuffer(shiftValue, sizeof(int64_t));
in HWTEST_F()
246 m_allTensors.emplace_back(shiftTensor);
in HWTEST_F()
359 std::shared_ptr<NNTensor> shiftTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim,
in HWTEST_F() local
361 m_allTensors.emplace_back(shiftTensor);
in HWTEST_F()
|
| H A D | pow_builder_test.cpp | 55 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, in HWTEST_F() local
181 shiftTensor->SetBuffer(shiftValue, sizeof(shiftValue)); in HWTEST_F()
182 m_allTensors.emplace_back(shiftTensor); in HWTEST_F()
252 std::shared_ptr<NNTensor> shiftTensor = TransToNNTensor(OH_NN_FLOAT32, m_shiftDim, in HWTEST_F() local
254 m_allTensors.emplace_back(shiftTensor); in HWTEST_F()
|
Completed in 2 milliseconds