Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/tests/
H A Dtensor_layout_test.cc220 Mesh reduced_mesh = replicated_layout.ReducedMesh(); in TEST_F() local
228 Mesh reduced_mesh = layout.ReducedMesh(); in TEST_F() local
241 Mesh reduced_mesh = layout->ReducedMesh(); in TEST_F() local
256 Mesh reduced_mesh = layout->ReducedMesh(); in TEST_F() local
273 Mesh reduced_mesh = layout->ReducedMesh(); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Dtensor_layout.cc748 Mesh reduced_mesh = ReducedAbstractMesh(this); in ReducedMesh() local
850 Mesh reduced_mesh = reduced_layout.mesh(); in HostShardMap() local