Home
last modified time | relevance | path

Searched refs:mesh_rank (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.tpu.experimental.-topology.pbtxt10 name: "mesh_rank"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.tpu.experimental.-topology.pbtxt10 name: "mesh_rank"
/external/tensorflow/tensorflow/python/tpu/
Ddevice_assignment.py97 if core_assignment.shape[-1] != topology.mesh_rank:
100 "rank ({}), got shape {}".format(topology.mesh_rank,
Dtopology.py155 def mesh_rank(self): member in Topology