Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_helpers.cc205 const std::vector<Device*>& host_devices, in ForConfigurationNodeMatchingType()
225 std::vector<Device*> host_devices; in ForConfigurationNodeMatchingType() local
H A Ddistributed_tpu_configuration_rewrite_pass.cc247 Graph* graph) -> Status { in Run()
365 Graph* graph) -> Status { in Run()
H A Dconfigure_tpu_embedding_rewrite_pass.cc219 Graph* graph) -> Status { in Run()