Python object for XLA compile options. These options can be passed to the 'compile' step when using a local XLA client.
| 503 | |
| 504 | |
| 505 | class CompileOptions(object): |
| 506 | """Python object for XLA compile options. |
| 507 | |
| 508 | These options can be passed to the 'compile' step when using a local XLA |
| 509 | client. |
| 510 | """ |
| 511 | |
| 512 | def __init__(self): |
| 513 | self.xla_dump_to = None |
| 514 | self.dump_hlo_pass_re = None |
| 515 | self.dump_hlo_module_re = None |
| 516 | self.dump_hlo_as_text = None |
| 517 | self.dump_hlo_as_proto = None |
| 518 | self.hlo_profile = None |
| 519 | self.num_replicas = 1 |
| 520 | self.num_partitions = 1 |
| 521 | self.argument_layouts = None |
| 522 | self.result_layout = None |
| 523 | self.device_assignment = None |
| 524 | |
| 525 | |
| 526 | class Computation(object): |