3#include "onnx_proto3.pb.h"
6namespace Experimental {
12 auto input_name = nodeproto.input(0);
16 throw std::runtime_error(
"TMVA::SOFIE ONNX Parser leaky relu op has input tensor" + input_name +
17 " but its type is not yet registered");
20 std::unique_ptr<ROperator> op;
22 float attr_alpha = 0.01;
24 for (
int_t i = 0; i < nodeproto.attribute_size(); i++) {
25 std::string attribute_name = nodeproto.attribute(i).name();
26 if (attribute_name ==
"alpha")
27 attr_alpha = nodeproto.attribute(i).f();
30 std::string output_name = nodeproto.output(0);
34 throw std::runtime_error(
"TMVA::SOFIE - Unsupported - Operator Leaky Relu does not yet support input type " +
35 std::to_string(
static_cast<int>(input_type)));
bool IsRegisteredTensorType(const std::string &)
void RegisterTensorType(const std::string &, ETensorType)
ETensorType GetTensorType(const std::string &name)
ParserFuncSignature ParseLeakyRelu
std::function< std::unique_ptr< ROperator >(RModelParser_ONNX &, const onnx::NodeProto &)> ParserFuncSignature
create variable transformations