Searched refs:tpu_device_names (Results 1 – 2 of 2) sorted by relevance
437 const std::vector<std::vector<string>>& tpu_device_names,
2894 const std::vector<std::vector<string>>& tpu_device_names, in BuildExecuteNodes() argument2899 TF_RET_CHECK(params_info.NumReplicas() == tpu_device_names.size()); in BuildExecuteNodes()3103 node->set_assigned_device_name(tpu_device_names[replica][core]); in BuildExecuteNodes()3240 tpu_device_names[replica][core], &device)); in BuildExecuteNodes()3846 const auto& tpu_device_names = iter->second; in LowerOutsideCompilationFunctionalNodes() local3851 TF_RET_CHECK(replica_id < tpu_device_names.size()); in LowerOutsideCompilationFunctionalNodes()3852 const string& tpu_device_name = tpu_device_names[replica_id][0]; in LowerOutsideCompilationFunctionalNodes()