Searched defs:start_device_id (Results 1 – 2 of 2) sorted by relevance
129 const int64_t start_device_id = mesh_config.min_global_device_id(); in UpdateTPUCompileMetadata() local
245 mlir::Value start_device_id = ops_util::GetR1Const( in LowerAllReduceOpImpl() local534 const mlir::Value start_device_id = ops_util::GetR1Const( in LowerAllGatherOp() local