diff options
Diffstat (limited to 'compiler/exo-tflite/src/Convert.cpp')
| -rw-r--r-- | compiler/exo-tflite/src/Convert.cpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/compiler/exo-tflite/src/Convert.cpp b/compiler/exo-tflite/src/Convert.cpp index 20035ffe9..56f3c63bb 100644 --- a/compiler/exo-tflite/src/Convert.cpp +++ b/compiler/exo-tflite/src/Convert.cpp @@ -55,18 +55,18 @@ void convert_to_TFLNodes(loco::Graph *graph) // Add converters for canonical nodes. Note: Not all loco canonical nodes are listed. - if (get_knob<Knob::ConvertAvgPool2D>()) + if (get<Knob::ConvertAvgPool2D>()) phase.emplace_back(stdex::make_unique<AvgPool2DConverter>()); // TODO loco::ConstGen // TODO loco::Conv2D // TODO loco::DepthwiseConv2D // TODO loco::DepthwiseFilterEncode - if (get_knob<Knob::ConvertEltwiseAdd>()) + if (get<Knob::ConvertEltwiseAdd>()) phase.emplace_back(stdex::make_unique<EltwiseAddConverter>()); // TODO loco::EltwiseDiv - if (get_knob<Knob::ConvertEltwiseMul>()) + if (get<Knob::ConvertEltwiseMul>()) phase.emplace_back(stdex::make_unique<EltwiseMulConverter>()); // TODO loco::EltwiseSqrt @@ -74,17 +74,17 @@ void convert_to_TFLNodes(loco::Graph *graph) // TODO loco::FeatureBiasAdd // TODO loco::FixedReshape - if (get_knob<Knob::ConvertMaxPool2D>()) + if (get<Knob::ConvertMaxPool2D>()) phase.emplace_back(stdex::make_unique<MaxPool2DConverter>()); - if (get_knob<Knob::ConvertRelu>()) + if (get<Knob::ConvertRelu>()) phase.emplace_back(stdex::make_unique<ReluConverter>()); // TODO loco::ReLU6 // TODO loco::Tanh // TODO loco::TensorConcat // TODO loco::TensorBiasAdd - if (get_knob<Knob::ConvertTensorBroadcast>()) + if (get<Knob::ConvertTensorBroadcast>()) phase.emplace_back(stdex::make_unique<TensorBroadcastConverter>()); // TODO loco::TensorSoftmax |