(op)
| 177 | |
| 178 | |
| 179 | def InferOpBlobDevices(op): |
| 180 | device_info = C.infer_op_input_output_device(op.SerializeToString()) |
| 181 | input_info = [] |
| 182 | output_info = [] |
| 183 | for dev_str in device_info[0]: |
| 184 | device_option = caffe2_pb2.DeviceOption() |
| 185 | device_option.ParseFromString(dev_str) |
| 186 | input_info.append(device_option) |
| 187 | for dev_str in device_info[1]: |
| 188 | device_option = caffe2_pb2.DeviceOption() |
| 189 | device_option.ParseFromString(dev_str) |
| 190 | output_info.append(device_option) |
| 191 | return input_info, output_info |
| 192 | |
| 193 | |
| 194 | def InferOpDeviceAsBlobDevices(op): |
no test coverage detected
searching dependent graphs…