Searched defs:topology_attr (Results 1 – 4 of 4) sorted by relevance
58 std::string topology_attr; in TEST_P() local299 std::string topology_attr = topology_proto.SerializeAsString(); in TEST() local322 std::string topology_attr; in TEST() local418 std::string topology_attr = topology_proto.SerializeAsString(); in TEST() local532 std::string topology_attr = topology_proto.SerializeAsString(); in TEST() local
261 llvm::StringRef topology_attr, int num_tasks, int num_tpus_per_task) { in ParseTopologyAttr()345 llvm::StringRef topology_attr, in GetGeneralTPUExecutionDeviceAssignment()447 llvm::StringRef topology_attr, in GetTPUCompilationAndExecutionDevices()502 auto topology_attr = in GetHostDeviceOutsideComputation() local
732 auto topology_attr = in Rewrite() local
1593 static Status ParseTopologyAttr(const string& topology_attr, in ParseTopologyAttr()1778 int num_cores_per_replica, const string& topology_attr, in BuildDeviceAssignment()