Home
last modified time | relevance | path

Searched refs:activationTensor (Results 1 - 3 of 3) sorted by relevance

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dmul_builder_test.cpp53 std::shared_ptr<NNTensor> activationTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
56 activationTensor->SetBuffer(activationValue, sizeof(int8_t)); in SaveParamsTensor()
57 m_allTensors.emplace_back(activationTensor); in SaveParamsTensor()
165 std::shared_ptr<NNTensor> activationTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, in HWTEST_F() local
168 activationTensor->SetBuffer(&activationValue, sizeof(activationValue)); in HWTEST_F()
169 m_allTensors.emplace_back(activationTensor); in HWTEST_F()
173 activationTensor->SetBuffer(nullptr, 0); in HWTEST_F()
188 std::shared_ptr<NNTensor> activationTensor = TransToNNTensor(OH_NN_INT8, m_paramDim, in HWTEST_F() local
191 activationTensor->SetBuffer(activationValue, 2 * sizeof(int8_t)); in HWTEST_F()
192 m_allTensors.emplace_back(activationTensor); in HWTEST_F()
209 std::shared_ptr<NNTensor> activationTensor = TransToNNTensor(OH_NN_INT8, m_paramDim, HWTEST_F() local
245 std::shared_ptr<NNTensor> activationTensor = TransToNNTensor(OH_NN_INT8, m_paramDim, HWTEST_F() local
[all...]
H A Dscale_builder_test.cpp64 std::shared_ptr<NNTensor> activationTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveActivationTensor() local
67 activationTensor->SetBuffer(activationValue, sizeof(int64_t)); in SaveActivationTensor()
68 m_allTensors.emplace_back(activationTensor); in SaveActivationTensor()
201 std::shared_ptr<NNTensor> activationTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, in HWTEST_F() local
204 activationTensor->SetBuffer(&activationValue, sizeof(activationValue)); in HWTEST_F()
205 m_allTensors.emplace_back(activationTensor); in HWTEST_F()
209 activationTensor->SetBuffer(nullptr, 0); in HWTEST_F()
248 std::shared_ptr<NNTensor> activationTensor = TransToNNTensor(OH_NN_INT8, in HWTEST_F() local
251 activationTensor->SetBuffer(activationValue, 2 * sizeof(int64_t)); in HWTEST_F()
252 m_allTensors.emplace_back(activationTensor); in HWTEST_F()
271 std::shared_ptr<NNTensor> activationTensor = TransToNNTensor(OH_NN_INT8, HWTEST_F() local
330 std::shared_ptr<NNTensor> activationTensor = TransToNNTensor(OH_NN_INT8, m_paramDim, HWTEST_F() local
[all...]
H A Dmatmul_builder_test.cpp80 std::shared_ptr<NNTensor> activationTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveActivationTensor() local
83 activationTensor->SetBuffer(activationValue, sizeof(int8_t)); in SaveActivationTensor()
84 m_allTensors.emplace_back(activationTensor); in SaveActivationTensor()
304 std::shared_ptr<NNTensor> activationTensor; in HWTEST_F() local
305 activationTensor = TransToNNTensor(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_MATMUL_ACTIVATION_TYPE); in HWTEST_F()
307 activationTensor->SetBuffer(&activationValue, sizeof(activationValue)); in HWTEST_F()
308 m_allTensors.emplace_back(activationTensor); in HWTEST_F()
312 activationTensor->SetBuffer(nullptr, 0); in HWTEST_F()
329 std::shared_ptr<NNTensor> activationTensor; in HWTEST_F() local
330 activationTensor in HWTEST_F()
352 std::shared_ptr<NNTensor> activationTensor; HWTEST_F() local
432 std::shared_ptr<NNTensor> activationTensor; HWTEST_F() local
[all...]

Completed in 3 milliseconds