(op)
| 192 | |
| 193 | |
| 194 | def InferOpDeviceAsBlobDevices(op): |
| 195 | op_dev = op.device_option if op.device_option else caffe2_pb2.DeviceOption() |
| 196 | input_dev = [op_dev] * len(op.input) |
| 197 | output_dev = [op_dev] * len(op.output) |
| 198 | return input_dev, output_dev |
| 199 | |
| 200 | |
| 201 | GradientSlice = namedtuple('GradientSlice', ['indices', 'values']) |
no outgoing calls
no test coverage detected
searching dependent graphs…