23 circle::BuiltinOperator opcode,
26 auto specific_configure_func = get_kernel_configure_func(opcode);
27 if (specific_configure_func ==
nullptr)
28 assert(
false &&
"Unsupported operator");
30 specific_configure_func(cur_op, runtime_graph);
34 circle::BuiltinOperator opcode,
37 auto specific_execute_func = get_kernel_execute_func(opcode);
38 if (specific_execute_func ==
nullptr)
39 assert(
false &&
"Unsupported operator");
41 specific_execute_func(cur_op, runtime_graph);
46training::Status training::KernelTrainRegistry::train_kernel(
47 const circle::Operator *cur_op, circle::BuiltinOperator opcode,
CircleReader *reader,
48 GradientCalculationStorage *gradient_calculation_storage,
const TrainingSettings &settings,
49 TrainableWeightStorage *weight_storage,
bool is_compute_gradient)
const
51 auto specific_train_func = get_kernel_train_func(opcode);
52 if (specific_train_func ==
nullptr)
53 assert(
false &&
"Unsupported operator");
55 return specific_train_func(cur_op, reader, gradient_calculation_storage, settings, weight_storage,
Loads Circle file and provides helpers to access attributes.
void execute_kernel(const circle::Operator *cur_op, circle::BuiltinOperator opcode, BaseRuntimeGraph *runtime_graph) const