Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dmul_builder_test.cpp56 activationTensor->SetBuffer(activationValue, sizeof(int8_t)); in SaveParamsTensor()
57 m_allTensors.emplace_back(activationTensor); in SaveParamsTensor()
168 activationTensor->SetBuffer(&activationValue, sizeof(activationValue));
169 m_allTensors.emplace_back(activationTensor);
173 activationTensor->SetBuffer(nullptr, 0);
191 activationTensor->SetBuffer(activationValue, 2 * sizeof(int8_t));
192 m_allTensors.emplace_back(activationTensor);
196 activationTensor->SetBuffer(nullptr, 0);
213 m_allTensors.emplace_back(activationTensor);
217 activationTensor->SetBuffer(nullptr, 0);
[all …]
H A Dscale_builder_test.cpp67 activationTensor->SetBuffer(activationValue, sizeof(int64_t)); in SaveActivationTensor()
68 m_allTensors.emplace_back(activationTensor); in SaveActivationTensor()
205 m_allTensors.emplace_back(activationTensor);
209 activationTensor->SetBuffer(nullptr, 0);
251 activationTensor->SetBuffer(activationValue, 2 * sizeof(int64_t));
252 m_allTensors.emplace_back(activationTensor);
256 activationTensor->SetBuffer(nullptr, 0);
275 m_allTensors.emplace_back(activationTensor);
279 activationTensor->SetBuffer(nullptr, 0);
332 m_allTensors.emplace_back(activationTensor);
[all …]
H A Dmatmul_builder_test.cpp84 m_allTensors.emplace_back(activationTensor); in SaveActivationTensor()
304 std::shared_ptr<NNTensor> activationTensor; variable
308 m_allTensors.emplace_back(activationTensor);
312 activationTensor->SetBuffer(nullptr, 0);
329 std::shared_ptr<NNTensor> activationTensor; variable
333 m_allTensors.emplace_back(activationTensor);
337 activationTensor->SetBuffer(nullptr, 0);
352 std::shared_ptr<NNTensor> activationTensor; variable
356 m_allTensors.emplace_back(activationTensor);
360 activationTensor->SetBuffer(nullptr, 0);
[all …]