Lines Matching refs:activation

41 namespace activation {  namespace
123 template <ActivationFn activation>
130     CalculateActivationRangeUint8(activation, inputShape, &output_activation_min,  in reluXQuant8()
218 template <ActivationFn activation>
225     CalculateActivationRangeInt8(activation, inputShape, &output_activation_min,  in reluXQuant8Signed()
620 NN_REGISTER_OPERATION(RELU, "RELU", std::bind(activation::validate, OperationType::RELU, _1),
621                       std::bind(activation::prepare, OperationType::RELU, _1),
622                       activation::executeRelu, .allowZeroSizedInput = true);
623 NN_REGISTER_OPERATION(RELU1, "RELU1", std::bind(activation::validate, OperationType::RELU1, _1),
624                       std::bind(activation::prepare, OperationType::RELU1, _1),
625                       activation::executeRelu1, .allowZeroSizedInput = true);
626 NN_REGISTER_OPERATION(RELU6, "RELU6", std::bind(activation::validate, OperationType::RELU6, _1),
627                       std::bind(activation::prepare, OperationType::RELU6, _1),
628                       activation::executeRelu6, .allowZeroSizedInput = true);
630                       std::bind(activation::validate, OperationType::LOGISTIC, _1),
631                       std::bind(activation::prepare, OperationType::LOGISTIC, _1),
632                       activation::executeLogistic, .allowZeroSizedInput = true);
633 NN_REGISTER_OPERATION(TANH, "TANH", std::bind(activation::validate, OperationType::TANH, _1),
634                       std::bind(activation::prepare, OperationType::TANH, _1),
635                       activation::executeTanh, .allowZeroSizedInput = true);
636 NN_REGISTER_OPERATION(HARD_SWISH, "HARD_SWISH", activation::validateHardSwish,
637                       std::bind(activation::prepare, OperationType::HARD_SWISH, _1),
638                       activation::executeHardSwish, .allowZeroSizedInput = true);