Searched refs:src_core_id (Results 1 – 2 of 2) sorted by relevance
515 auto src_core_id = static_cast<const PodBufferHandle*>(src)->core_id(); in TransferFromDeviceToDevice() local518 auto src_driver_id = core_to_driver_id_[src_core_id]; in TransferFromDeviceToDevice()
1061 auto src_core_id = state_.b->AddInstruction(HloInstruction::CreateConstant( in Broadcast() local1067 ShapeUtil::MakeShape(PRED, {}), state_.partition_id, src_core_id, in Broadcast()