Searched defs:SpmdPartitionerOptions (Results 1 – 1 of 1) sorted by relevance
38 struct SpmdPartitionerOptions { struct41 bool conv_halo_exchange_always_on_lhs = true;45 int64_t report_instruction_count = 5;49 int64_t threshold_for_windowed_einsum_mib = 256;52 bool unroll_windowed_einsum = false;55 bool bidirectional_windowed_einsum = false;58 bool allow_module_signature_change = false;64 bool cache_all_gather = true;68 bool choose_faster_windowed_einsum_over_mem = false;72 bool bidirectional_decomposed_all_gather = false;