Searched refs:activationTensor (Results 1 - 3 of 3) sorted by relevance
/foundation/ai/neural_network_runtime/test/unittest/ops/ |
H A D | mul_builder_test.cpp | 53 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 D | scale_builder_test.cpp | 64 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 D | matmul_builder_test.cpp | 80 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