Searched defs:tpu_replicate_device_names_mapping (Results 1 – 1 of 1) sorted by relevance
3713 const TPUReplicateDeviceNamesMapping& tpu_replicate_device_names_mapping) { in LowerOutsideCompilationFunctionalNodes()4217 TPUReplicateDeviceNamesMapping* tpu_replicate_device_names_mapping, in RewriteTPUReplicateNode()4478 TPUReplicateDeviceNamesMapping tpu_replicate_device_names_mapping; in Run() local