Builders.h File Reference

#include "luci_interpreter/core/reader/CircleMicroReader.h"
#include "core/RuntimeModule.h"
#include "KernelsToBuild.lst"

Go to the source code of this file.

Namespaces
namespace	luci_interpreter

Macros
#define	REGISTER_KERNEL(builtin_operator, name)

Macro Definition Documentation

REGISTER_KERNEL

#define REGISTER_KERNEL	(		builtin_operator,
			name
	)

Value:

  void configure_kernel_Circle##name(const circle::Operator *cur_op,   \
                                     BaseRuntimeGraph *runtime_graph); \
                                                                       \
  void execute_kernel_Circle##name(const circle::Operator *cur_op, BaseRuntimeGraph *runtime_graph);

Definition at line 35 of file Builders.h.

{
#define REGISTER_TRAIN_KERNEL(builtin_operator, name)                                           \
  Status train_kernel_Circle##name(                                                             \
    const circle::Operator *op, CircleReader *reader,                                           \
    GradientCalculationStorage *gradient_calculation_storage, const TrainingSettings &settings, \
    TrainableWeightStorage *weight_storage, bool is_compute_gradient);
 
#include "KernelsToTrain.lst"
 
#undef REGISTER_TRAIN_KERNEL
} // namespace training
#endif // ENABLE_TRAINING
 
} // namespace luci_interpreter
 
#endif // LUCI_INTERPRETER_KERNELS_NODES_BUILDERS_H