Searched defs:tpu_replicate_device_names_mapping (Results 1 – 1 of 1) sorted by relevance
4167 const TPUReplicateDeviceNamesMapping& tpu_replicate_device_names_mapping) { in LowerOutsideCompilationFunctionalNodes()4718 TPUReplicateDeviceNamesMapping* tpu_replicate_device_names_mapping, in RewriteTPUReplicateNode()5004 TPUReplicateDeviceNamesMapping tpu_replicate_device_names_mapping; in InternalRun() local