#include "tensorflow/lite/kernels/internal/reference/broadcast_args.h"
#include <algorithm>
#include <cstdint>
#include <memory>
#include "tensorflow/lite/core/c/common.h"
#include "tensorflow/lite/kernels/internal/tensor.h"
#include "tensorflow/lite/kernels/kernel_util.h"
namespace tflite {
namespace ops {
namespace builtin {
namespace broadcast_args {
constexpr int kShape1Tensor = …;
constexpr int kShape2Tensor = …;
constexpr int kOutputTensor = …;
struct BroadcastArgsContext { … };
TfLiteStatus EvalImpl(TfLiteContext* context, TfLiteNode* node);
TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { … }
TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { … }
TfLiteStatus EvalImpl(TfLiteContext* context, TfLiteNode* node) { … }
}
TfLiteRegistration* Register_BROADCAST_ARGS() { … }
}
}
}