Searched defs:enable_spmd_xla_partitioning (Results 1 – 1 of 1) sorted by relevance
1228 bool enable_spmd_xla_partitioning = false; in ComputeAsync() local1275 bool enable_spmd_xla_partitioning = false; in ComputeAsync() local1636 bool enable_spmd_xla_partitioning, const TPUMetadata& tpu_metadata) { in ReplaceResourceArgsWithVarHandleOps()2182 Graph* graph, bool enable_spmd_xla_partitioning, int num_cores_per_replica, in OptimizeTpuInputOutputTensors()