Searched refs:m_paramsDim (Results 1 – 1 of 1) sorted by relevance
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | gelu_builder_test.cpp | 43 std::vector<int32_t> m_paramsDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::GeluBuilderTest
69 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_GELU_APPROXIMATE);
84 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_GELU_APPROXIMATE);
104 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_GELU_APPROXIMATE);
122 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_GELU_APPROXIMATE);
167 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_GELU_APPROXIMATE);
183 std::shared_ptr<NNTensor> approximateTensor = TransToNNTensor(OH_NN_INT64, m_paramsDim,
203 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
219 std::shared_ptr<NNTensor> approximateTensor = TransToNNTensor(OH_NN_BOOL, m_paramsDim,
236 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_GELU_APPROXIMATE);
|