Home
last modified time | relevance | path

Searched refs:ParseTopologyAttr (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc264 StatusOr<xla::Array4D<TaskAndDevice>> ParseTopologyAttr( in ParseTopologyAttr() function
354 TF_ASSIGN_OR_RETURN(auto topology, ParseTopologyAttr(topology_attr, num_tasks, in GetGeneralTPUExecutionDeviceAssignment()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc1384 static Status ParseTopologyAttr(const string& topology_attr, in ParseTopologyAttr() function
1647 TF_RETURN_IF_ERROR(ParseTopologyAttr(topology_attr, tpu_topology, num_tasks, in BuildDeviceAssignment()