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()506 auto topology_attr = in GetHostDeviceOutsideComputation() local
571 auto topology_attr = in Rewrite() local
1647 static Status ParseTopologyAttr(const string& topology_attr, in ParseTopologyAttr()1831 int num_cores_per_replica, const string& topology_attr, in BuildDeviceAssignment()