operands.iterate([&](const operand::Index &index, const operand::Object &object) {
bool showing_cond = false;
-
+ auto usage = object.getUsage();
if (option == OPTIONS::SHOW_CONSTANTS)
{
showing_cond = object.usageIsDefined();
}
else
{
- auto usage = object.getUsage();
showing_cond = (usage == operand::OperandUsage::MODEL_INPUT) ||
(usage == operand::OperandUsage::OPERATION_OUTPUT);
}
- showing_cond = showing_cond && (object.getUses().size() > 0);
+ if (usage != operand::OperandUsage::OPERATION_OUTPUT)
+ {
+ showing_cond = showing_cond && (object.getUses().size() > 0);
+ }
if (showing_cond)
{
neurun::dumper::dot::DotOperandInfo operand_info(index, object);
// Generate output operand and permute operation
auto out_operand_index = addOperand(operand.shape(), operand.typeInfo());
auto &out_operand = _operands.at(out_operand_index);
+ out_operand.setAsOperationOutput();
// change model output if operand_index is model output index
auto &model_outputs = getOutputs();
if (model_outputs.contains(operand_index))