| 10 | |
| 11 | namespace { |
| 12 | void output_info(std::shared_ptr<Network> network, size_t output_size) { |
| 13 | for (size_t index = 0; index < output_size; index++) { |
| 14 | printf("output[%zu] names %s \n", index, |
| 15 | network->get_all_output_name()[index].c_str()); |
| 16 | std::shared_ptr<Tensor> output_tensor = network->get_output_tensor(index); |
| 17 | size_t ndim = output_tensor->get_layout().ndim; |
| 18 | for (size_t i = 0; i < ndim; i++) { |
| 19 | printf("output[%zu] tensor.shape[%zu] %zu \n", index, i, |
| 20 | output_tensor->get_layout().shapes[i]); |
| 21 | } |
| 22 | } |
| 23 | } |
| 24 | |
| 25 | void output_data_info(std::shared_ptr<Network> network, size_t output_size) { |
| 26 | for (size_t index = 0; index < output_size; index++) { |
no test coverage detected