einsum.cpp

// Copyright (C) 2018-2022 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//

#include "op_table.hpp"
#include "openvino/opsets/opset8.hpp"

using namespace std;
using namespace ov::opset8;

namespace ov {
namespace frontend {
namespace tensorflow {
namespace op {
OutputVector translate_einsum_op(const NodeContext& node) {
    auto op_type = node.get_op_type();
    TENSORFLOW_OP_VALIDATION(node, op_type == "Einsum", "Internal error: incorrect usage of translate_einsum_op.");
    auto equation = node.get_attribute<std::string>("equation");

    OutputVector inputs;
    for (size_t input_ind = 0; input_ind < node.get_input_size(); ++input_ind) {
        inputs.push_back(node.get_input(input_ind));
    }

    auto einsum = make_shared<Einsum>(inputs, equation);
    set_node_name(node.get_name(), einsum);
    return {einsum};
}
}  // namespace op
}  // namespace tensorflow
}  // namespace frontend
}  // namespace ov