Searched defs:SpmdPartitionerOptions (Results 1 – 1 of 1) sorted by relevance
36 struct SpmdPartitionerOptions { struct39 bool conv_halo_exchange_always_on_lhs = true;43 int64 report_instruction_count = 5;47 int64 threshold_for_windowed_einsum_mib = 256;50 bool unroll_windowed_einsum = false;53 bool bidirectional_windowed_einsum = false;56 bool allow_module_signature_change = false;62 bool cache_all_gather = true;65 bool choose_faster_windowed_einsum_over_mem = false;