Home
last modified time | relevance | path

Searched refs:TPUReplicateDeviceNamesMapping (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.h250 TPUReplicateDeviceNamesMapping; typedef
510 const TPUReplicateDeviceNamesMapping& tpu_replicate_device_names_mapping);
558 TPUReplicateDeviceNamesMapping* tpu_replicate_device_names_mapping,
Ddistributed_tpu_rewrite_pass.cc3713 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()