Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc693 const Mesh& target_mesh = target_layout.mesh(); in CopyToMesh() local
821 const Mesh& target_mesh = target_layout->mesh(); in Pack() local
925 const Mesh& target_mesh = target_layout->mesh(); in SparsePack() local
H A Ddtensor_device_util.cc62 const Mesh& target_mesh = mesh.mesh_config(); in BroadcastTensorHandleToParallelTensor() local