Searched refs:activation_ (Results 1 – 8 of 8) sorted by relevance
92 activation_ = ::tflite::ActivationFunctionType_RELU; in ReluActivation()97 activation_ = ::tflite::ActivationFunctionType_RELU6; in Relu6Activation()102 activation_ = ::tflite::ActivationFunctionType_RELU_N1_TO_1; in ReluMinus1To1Activation()107 activation_ = ::tflite::ActivationFunctionType_TANH; in TanhActivation()112 activation_ = ::tflite::ActivationFunctionType_SIGN_BIT; in SignBitActivation()122 return activation_; in Activation()133 ::tflite::ActivationFunctionType activation_ =
126 activation_ = ::tflite::ActivationFunctionType_RELU; in ReluActivation()131 activation_ = ::tflite::ActivationFunctionType_RELU6; in Relu6Activation()136 activation_ = ::tflite::ActivationFunctionType_RELU_N1_TO_1; in ReluMinus1To1Activation()141 activation_ = ::tflite::ActivationFunctionType_TANH; in TanhActivation()146 activation_ = ::tflite::ActivationFunctionType_SIGN_BIT; in SignBitActivation()158 return activation_; in Activation()170 ::tflite::ActivationFunctionType activation_ =
176 activation_ = ::tflite::ActivationFunctionType_RELU; in ReluActivation()181 activation_ = ::tflite::ActivationFunctionType_RELU6; in Relu6Activation()186 activation_ = ::tflite::ActivationFunctionType_RELU_N1_TO_1; in ReluMinus1To1Activation()191 activation_ = ::tflite::ActivationFunctionType_TANH; in TanhActivation()196 activation_ = ::tflite::ActivationFunctionType_SIGN_BIT; in SignBitActivation()208 return activation_; in Activation()225 ::tflite::ActivationFunctionType activation_ =
180 activation_ = ::tflite::ActivationFunctionType_RELU; in ReluActivation()185 activation_ = ::tflite::ActivationFunctionType_RELU6; in Relu6Activation()190 activation_ = ::tflite::ActivationFunctionType_RELU_N1_TO_1; in ReluMinus1To1Activation()195 activation_ = ::tflite::ActivationFunctionType_TANH; in TanhActivation()200 activation_ = ::tflite::ActivationFunctionType_SIGN_BIT; in SignBitActivation()212 return activation_; in Activation()229 ::tflite::ActivationFunctionType activation_ =
82 activation_ = ::tflite::ActivationFunctionType_RELU; in ReluActivation()87 activation_ = ::tflite::ActivationFunctionType_RELU6; in Relu6Activation()92 activation_ = ::tflite::ActivationFunctionType_RELU_N1_TO_1; in ReluMinus1To1Activation()102 return activation_; in Activation()113 ::tflite::ActivationFunctionType activation_ =
34 switch (activation_) { in Build()59 fprintf(stderr, "Activation %d is not supported.\n", activation_); in Build()66 if (!(activation_ == kTfLiteActRelu6 || activation_ == kTfLiteActReluN1To1)) { in PopulateSubgraph()73 const int relu_threshold = activation_ == kTfLiteActRelu6 ? 6 : 1; in PopulateSubgraph()
32 : OpBuilder(graph_builder), activation_(activation) {} in ActivationLayerBuilder()39 activation_ = activation; in SetActivation()53 TfLiteFusedActivation activation_;
497 PackWeightToTensor(&activation_tensor_, activation_, activation_size_); in GetActivation()498 activation_tensor_.data.int8 = activation_.data(); in GetActivation()566 std::vector<int8_t> activation_; member in tflite::__anon39d466b60111::QuantizedLstmParam