Skip to content

Expose model inputs and outputs number in AI.MODELGET #529

Closed
@DvirDukhan

Description

@DvirDukhan

As INPUTS and OUTPUTS are required only for TF models in the AI.MODELSET command, we want to allow investigate model's input and outputs:

  • ONNX:
 size_t n_input_nodes;
 ort->SessionGetInputCount(session, &n_input_nodes);
 size_t n_output_nodes;
 ort->SessionGetOutputCount(session, &n_output_nodes);
  • Torch - User defined

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions