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