Home
last modified time | relevance | path

Searched refs:allow_module_signature_change (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h56 bool allow_module_signature_change = false; member
Dspmd_partitioner.cc3687 if (!options_.allow_module_signature_change) { in Run()
3780 if (!options_.allow_module_signature_change) { in PreprocessSharding()
Dspmd_partitioner_test.cc46 options.allow_module_signature_change = true; in PartitionComputation()