Searched refs:ParseTopologyAttr (Results 1 – 2 of 2) sorted by relevance
260 StatusOr<xla::Array4D<TaskAndDevice>> ParseTopologyAttr( in ParseTopologyAttr() function350 TF_ASSIGN_OR_RETURN(auto topology, ParseTopologyAttr(topology_attr, num_tasks, in GetGeneralTPUExecutionDeviceAssignment()
1593 static Status ParseTopologyAttr(const string& topology_attr, in ParseTopologyAttr() function1860 TF_RETURN_IF_ERROR(ParseTopologyAttr(topology_attr, tpu_topology, num_tasks, in BuildDeviceAssignment()