Home
last modified time | relevance | path

Searched refs:epsilonTensor (Results 1 – 4 of 4) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dbatchnorm_builder_test.cpp56 epsilonTensor->SetBuffer(epsilonValue, sizeof(float)); in SaveParamsTensor()
57 m_allTensors.emplace_back(epsilonTensor); in SaveParamsTensor()
161 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_INT32, m_paramDim, variable
164 epsilonTensor->SetBuffer(&epsilonValue, sizeof(epsilonValue));
165 m_allTensors.emplace_back(epsilonTensor);
169 epsilonTensor->SetBuffer(nullptr, 0);
183 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
186 epsilonTensor->SetBuffer(epsilonValue, 2 * sizeof(float));
187 m_allTensors.emplace_back(epsilonTensor);
191 epsilonTensor->SetBuffer(nullptr, 0);
[all …]
H A Dinstance_norm_test.cpp55 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
58 epsilonTensor->SetBuffer(epsilonValue, sizeof(float)); in SaveParamsTensor()
59 m_allTensors.emplace_back(epsilonTensor); in SaveParamsTensor()
180 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
184 epsilonTensor->SetBuffer(epsilonValue, sizeof(epsilonValue));
185 m_allTensors.emplace_back(epsilonTensor);
216 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
218 m_allTensors.emplace_back(epsilonTensor);
H A Dlayernorm_builder_test.cpp262 std::shared_ptr<NNTensor> epsilonTensor; variable
263 epsilonTensor = TransToNNTensor(OH_NN_INT32, m_paramDim, nullptr, OH_NN_LAYER_NORM_EPSILON);
265 epsilonTensor->SetBuffer(&epsilonValue, sizeof(epsilonValue));
266 m_allTensors.emplace_back(epsilonTensor);
270 epsilonTensor->SetBuffer(nullptr, 0);
287 std::shared_ptr<NNTensor> epsilonTensor; variable
290 epsilonTensor->SetBuffer(epsilonValue, 2 * sizeof(float));
291 m_allTensors.emplace_back(epsilonTensor);
295 epsilonTensor->SetBuffer(nullptr, 0);
395 std::shared_ptr<NNTensor> epsilonTensor; variable
[all …]
H A Dl2_normalize_test.cpp65 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetEpsilon() local
68 epsilonTensor->SetBuffer(epsilonValue, sizeof(float)); in SetEpsilon()
69 m_allTensors.emplace_back(epsilonTensor); in SetEpsilon()
205 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
208 epsilonTensor->SetBuffer(epsilonValue, sizeof(int64_t));
209 m_allTensors.emplace_back(epsilonTensor);
320 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
322 m_allTensors.emplace_back(epsilonTensor);