Home
last modified time | relevance | path

Searched defs:configuration_device_name (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dconfigure_tpu_embedding_rewrite_pass.cc102 Status AddExecutePartitionerNode(const string& configuration_device_name, in AddExecutePartitionerNode()
136 Status AddCollateMemoryNode(const string& configuration_device_name, in AddCollateMemoryNode()
177 Status AddFinalizeNode(const string& configuration_device_name, in AddFinalizeNode()
219 Graph* graph) -> Status { in Run()
H A Ddistributed_tpu_configuration_rewrite_pass.cc58 Status AddConfigurationNode(const string& configuration_device_name, in AddConfigurationNode()
98 Status AddWaitNode(const string& configuration_device_name, in AddWaitNode()
247 Graph* graph) -> Status { in Run()
H A Ddistributed_tpu_rewrite_helpers.cc204 const string& configuration_device_name, in ForConfigurationNodeMatchingType()