(op)
| 164 | |
| 165 | |
| 166 | def InferOpBlobDevicesAsDict(op): |
| 167 | input_dev_list, output_dev_list = InferOpBlobDevices(op) |
| 168 | input_dict = { |
| 169 | op.input[i]: input_dev_list[i] |
| 170 | for i in range(len(op.input)) |
| 171 | } |
| 172 | output_dict = { |
| 173 | op.output[i]: output_dev_list[i] |
| 174 | for i in range(len(op.output)) |
| 175 | } |
| 176 | return input_dict, output_dict |
| 177 | |
| 178 | |
| 179 | def InferOpBlobDevices(op): |
no test coverage detected
searching dependent graphs…