Searched defs:topology_attr (Results 1 – 4 of 4) sorted by relevance
57 std::string topology_attr; in TEST_P() local298 std::string topology_attr = topology_proto.SerializeAsString(); in TEST() local321 std::string topology_attr; in TEST() local417 std::string topology_attr = topology_proto.SerializeAsString(); in TEST() local531 std::string topology_attr = topology_proto.SerializeAsString(); in TEST() local
265 llvm::StringRef topology_attr, int num_tasks, int num_tpus_per_task) { in ParseTopologyAttr()349 llvm::StringRef topology_attr, in GetGeneralTPUExecutionDeviceAssignment()451 llvm::StringRef topology_attr, in GetTPUCompilationAndExecutionDevices()508 auto topology_attr = in GetHostDeviceOutsideComputation() local
584 auto topology_attr = in Rewrite() local
1384 static Status ParseTopologyAttr(const string& topology_attr, in ParseTopologyAttr()1564 int num_cores_per_replica, const string& topology_attr, in BuildDeviceAssignment()