Home
last modified time | relevance | path

Searched defs:mesh_dim (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util_test.cc116 for (int mesh_dim : mesh_shape) topology_proto.add_mesh_shape(mesh_dim); in TopologyWithMeshShape() local
124 for (int mesh_dim : mesh_shape) topology_proto.add_mesh_shape(mesh_dim); in TopologyWithMeshShapeAndTasks() local