Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dpywrap_dtensor_device.cc106 bool is_host_mesh) { in PYBIND11_MODULE()
145 const std::string& serialized_mesh) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc2022 void AddMesh(const std::string& serialized_mesh, void* device_info, in AddMesh()
2070 void ExperimentalSetDefaultMesh(const std::string& serialized_mesh, in ExperimentalSetDefaultMesh()
H A Ddtensor_device_util.cc854 std::string serialized_mesh; in IdentifyAllFunctionsToExecute() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dspmd_expander_common.cc488 string serialized_mesh = mesh->ToString(); in GetMeshCoordinatesFromCluster() local