get the output tensor name in the order of graph
| 1111 | |
| 1112 | //! get the output tensor name in the order of graph |
| 1113 | const char* NetworkImplDft::get_output_name(size_t index) const { |
| 1114 | LITE_ASSERT( |
| 1115 | index < m_load_result.output_var_list.size(), |
| 1116 | "The output tensor index is large than the total outputs number."); |
| 1117 | return m_load_result.output_var_list[index].node()->name().c_str(); |
| 1118 | } |
| 1119 | |
| 1120 | //! get the input tensor name in the order of graph |
| 1121 | const char* NetworkImplDft::get_input_name(size_t index) const { |