Searched defs:epsilonTensor (Results 1 – 4 of 4) sorted by relevance
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
H A D | batchnorm_builder_test.cpp | 53 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local 161 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_INT32, m_paramDim, variable 183 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable 218 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, nullptr, variable
|
H A D | instance_norm_test.cpp | 55 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local 180 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable 216 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
|
H A D | l2_normalize_test.cpp | 65 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetEpsilon() local 205 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable 320 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
|
H A D | layernorm_builder_test.cpp | 262 std::shared_ptr<NNTensor> epsilonTensor; variable 287 std::shared_ptr<NNTensor> epsilonTensor; variable 395 std::shared_ptr<NNTensor> epsilonTensor; variable
|