Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/
H A Dupdate_tpu_metadata.cc114 std::string mesh_name = mesh_config.name(); in UpdateTPUCompileMetadata() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/python/
H A Ddtensor_device.py328 def set_tpu_core_ids(self, mesh_name, tpu_core_ids): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dpywrap_dtensor_device.cc175 const std::vector<int>& tpu_core_ids) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc161 Status SetTPUCoreIDs(const std::string& mesh_name, in SetTPUCoreIDs()
2090 void SetTPUCoreIDs(const std::string& mesh_name, in SetTPUCoreIDs()