Home
last modified time | relevance | path

Searched refs:activationTypeTensor (Results 1 - 2 of 2) sorted by relevance

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dsub_builder_test.cpp44 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 Dl2_normalize_test.cpp228 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