Home
last modified time | relevance | path

Searched refs:FusedActivationFunc (Results 1 – 10 of 10) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DBroadcast.cpp58 case static_cast<int32_t>(FusedActivationFunc::NONE): \
61 case static_cast<int32_t>(FusedActivationFunc::RELU): \
64 case static_cast<int32_t>(FusedActivationFunc::RELU1): \
67 case static_cast<int32_t>(FusedActivationFunc::RELU6): \
213 NN_RET_CHECK_EQ(static_cast<FusedActivationFunc>(activation), FusedActivationFunc::NONE); in executeInt32()
/packages/modules/NeuralNetworks/common/types/src/
DTypeUtils.cpp489 std::ostream& operator<<(std::ostream& os, const FusedActivationFunc& activation) { in operator <<()
491 case FusedActivationFunc::NONE: in operator <<()
493 case FusedActivationFunc::RELU: in operator <<()
495 case FusedActivationFunc::RELU1: in operator <<()
497 case FusedActivationFunc::RELU6: in operator <<()
DValidation.cpp225 Result<Version> validateFusedActivationFunc(const FusedActivationFunc& activation) { in validateFusedActivationFunc()
227 case FusedActivationFunc::NONE: in validateFusedActivationFunc()
228 case FusedActivationFunc::RELU: in validateFusedActivationFunc()
229 case FusedActivationFunc::RELU1: in validateFusedActivationFunc()
230 case FusedActivationFunc::RELU6: in validateFusedActivationFunc()
1854 Result<Version> validate(const FusedActivationFunc& activation) { in validate()
/packages/modules/NeuralNetworks/common/types/include/nnapi/
DValidation.h52 Result<Version> validate(const FusedActivationFunc& activation);
DTypes.h194 enum class FusedActivationFunc : int32_t { enum
DTypeUtils.h99 std::ostream& operator<<(std::ostream& os, const FusedActivationFunc& activation);
/packages/modules/NeuralNetworks/runtime/
DNeuralNetworks.cpp351 static_assert(static_cast<int32_t>(FusedActivationFunc::NONE) == ANEURALNETWORKS_FUSED_NONE,
353 static_assert(static_cast<int32_t>(FusedActivationFunc::RELU) == ANEURALNETWORKS_FUSED_RELU,
355 static_assert(static_cast<int32_t>(FusedActivationFunc::RELU1) == ANEURALNETWORKS_FUSED_RELU1,
357 static_assert(static_cast<int32_t>(FusedActivationFunc::RELU6) == ANEURALNETWORKS_FUSED_RELU6,
/packages/modules/NeuralNetworks/tools/api/
Dtypes.spec13 %define FusedActivationFunc FuseCode
71 %define FusedActivationFunc FusedActivationFunc
426 * {@link %{FusedActivationFunc}} values. Specifies the activation to
430 * the {@link %{FusedActivationFunc}} must be "NONE".
495 * {@link %{FusedActivationFunc}} values. Specifies the activation to
519 * {@link %{FusedActivationFunc}} values. Specifies the activation to
1184 * {@link %{FusedActivationFunc}} values. Specifies the activation to
1361 * {@link %{FusedActivationFunc}} values. Specifies the activation to
1385 * {@link %{FusedActivationFunc}} values. Specifies the activation to
1855 * {@link %{FusedActivationFunc}} values. Specifies the activation to
[all …]
DTypes.t125 %insert FusedActivationFunc
DNeuralNetworksTypes.t92 %insert FusedActivationFunc