Searched refs:activationTypeTensor (Results 1 - 2 of 2) sorted by relevance
/foundation/ai/neural_network_runtime/test/unittest/ops/ |
H A D | sub_builder_test.cpp | 44 std::shared_ptr<NNTensor> activationTypeTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveActivateTypeTensor() local 47 activationTypeTensor->SetBuffer(activationTypeValue, sizeof(int8_t)); in SaveActivateTypeTensor() 48 m_allTensors.emplace_back(activationTypeTensor); in SaveActivateTypeTensor() 218 std::shared_ptr<NNTensor> activationTypeTensor = TransToNNTensor(OH_NN_INT8, paramDim, nullptr, in HWTEST_F() local 220 activationTypeTensor->SetBuffer(nullptr, 0); in HWTEST_F() 221 m_allTensors.emplace_back(activationTypeTensor); in HWTEST_F() 276 std::shared_ptr<NNTensor> activationTypeTensor = TransToNNTensor(OH_NN_INT8, paramDim, nullptr, in HWTEST_F() local 280 activationTypeTensor->SetBuffer(activationTypeValue, sizeof(int8_t)); in HWTEST_F() 281 m_allTensors.emplace_back(activationTypeTensor); in HWTEST_F()
|
H A D | l2_normalize_test.cpp | 228 std::shared_ptr<NNTensor> activationTypeTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim,
in HWTEST_F() local 231 activationTypeTensor->SetBuffer(activationTypeValue, sizeof(float));
in HWTEST_F() 232 m_allTensors.emplace_back(activationTypeTensor);
in HWTEST_F()
|
Completed in 2 milliseconds