Searched defs:use_spmd_for_xla_partitioning (Results 1 – 5 of 5) sorted by relevance
121 bool use_spmd_for_xla_partitioning = 15; field
570 const std::string& use_spmd_for_xla_partitioning, in UpdateTfMlirBridgeGraphAnalysisPerOp()
907 use_spmd_for_xla_partitioning=True, argument
847 use_spmd_for_xla_partitioning=False): argument
2292 Graph* graph, int device_ordinal, bool* use_spmd_for_xla_partitioning, in GetGraphFromFunction()