Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/
H A Dcollective_lowering.cc245 mlir::Value start_device_id = ops_util::GetR1Const( in LowerAllReduceOpImpl() local
534 const mlir::Value start_device_id = ops_util::GetR1Const( in LowerAllGatherOp() local
H A Dupdate_tpu_metadata.cc129 const int64_t start_device_id = mesh_config.min_global_device_id(); in UpdateTPUCompileMetadata() local