Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit c43d55f0 authored by Siarhei Vishniakou's avatar Siarhei Vishniakou Committed by Android (Google) Code Review
Browse files

Merge "Specify model name"

parents 8f41d27b 11acb738
Loading
Loading
Loading
Loading
+8 −2
Original line number Diff line number Diff line
@@ -512,6 +512,13 @@ std::string SlotState::dump() const {
    return out;
}

class AndroidPalmRejectionModel : public ::ui::OneDeviceTrainNeuralStylusPalmDetectionFilterModel {
public:
    AndroidPalmRejectionModel()
          : ::ui::OneDeviceTrainNeuralStylusPalmDetectionFilterModel(/*default version*/ "",
                                                                     std::vector<float>()) {}
};

PalmRejector::PalmRejector(const AndroidPalmFilterDeviceInfo& info,
                           std::unique_ptr<::ui::PalmDetectionFilter> filter)
      : mSharedPalmState(std::make_unique<::ui::SharedPalmDetectionFilterState>()),
@@ -523,8 +530,7 @@ PalmRejector::PalmRejector(const AndroidPalmFilterDeviceInfo& info,
        return;
    }
    std::unique_ptr<::ui::NeuralStylusPalmDetectionFilterModel> model =
            std::make_unique<::ui::OneDeviceTrainNeuralStylusPalmDetectionFilterModel>(
                    std::vector<float>());
            std::make_unique<AndroidPalmRejectionModel>();
    mPalmDetectionFilter =
            std::make_unique<::ui::NeuralStylusPalmDetectionFilter>(mDeviceInfo, std::move(model),
                                                                    mSharedPalmState.get());