Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc3079 const std::vector<std::vector<string>>& tpu_device_names, in CreatePartitionedDummyVarArgs()
3247 const std::vector<std::vector<string>>& tpu_device_names, in BuildExecuteNodes()
4300 const auto& tpu_device_names = iter->second; in LowerOutsideCompilationFunctionalNodes() local