auto operation_index = _graph.operands().at(index).getUses().list().front();
auto operation = &_graph.operations().at(operation_index);
auto fc_operation =
- dynamic_cast<const neurun::graph::operation::FullyConnected::Node *>(operation);
+ dynamic_cast<const neurun::graph::operation::FullyConnectedNode *>(operation);
if (fc_operation == nullptr)
break;
- auto ifm_index =
- fc_operation->getInputs().at(neurun::graph::operation::FullyConnected::Input::INPUT);
+ auto ifm_index = fc_operation->getInputs().at(
+ neurun::graph::operation::FullyConnectedNode::Input::INPUT);
const auto &ifm = _graph.operands().at(ifm_index);
const auto ifm_shape = ifm.shape().asFeature();
const auto num_output = shape.dim(0);