chromium/services/passage_embeddings/passage_embedder_execution_task.cc

// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "services/passage_embeddings/passage_embedder_execution_task.h"

#include "base/check_op.h"
#include "third_party/tflite_support/src/tensorflow_lite_support/cc/task/core/task_utils.h"

namespace passage_embeddings {

PassageEmbedderExecutionTask::PassageEmbedderExecutionTask(
    std::unique_ptr<tflite::task::core::TfLiteEngine> tflite_engine)
    : … { … }

PassageEmbedderExecutionTask::~PassageEmbedderExecutionTask() { … }

std::optional<OutputType> PassageEmbedderExecutionTask::Execute(
    InputType input) { … }

absl::Status PassageEmbedderExecutionTask::Preprocess(
    const std::vector<TfLiteTensor*>& input_tensors,
    InputType input) { … }

tflite::support::StatusOr<OutputType> PassageEmbedderExecutionTask::Postprocess(
    const std::vector<const TfLiteTensor*>& output_tensors,
    InputType input) { … }

}  // namespace passage_embeddings