Searched refs:restrict_functionalization_to_tpu_nodes (Results 1 – 5 of 5) sorted by relevance
181 bool restrict_functionalization_to_tpu_nodes) { in UpgradeLegacyGraph() argument185 restrict_functionalization_to_tpu_nodes in UpgradeLegacyGraph()
42 bool restrict_functionalization_to_tpu_nodes);
65 bool restrict_functionalization_to_tpu_nodes = false; member
3896 specs.restrict_functionalization_to_tpu_nodes)); in ConvertGraphToMlir()
357 import_config.restrict_functionalization_to_tpu_nodes = true; in Run()