Searched refs:GetInstructionDevice (Results 1 – 2 of 2) sorted by relevance
160 auto device = GetInstructionDevice(*comm_instruction); in VerifyCompanionSets()281 absl::optional<int64> HloModuleGroupMetadata::GetInstructionDevice( in GetInstructionDevice() function in xla::HloModuleGroupMetadata452 auto send_device = GetInstructionDevice(*channel.send); in VerifyChannelInstructions()453 auto send_done_device = GetInstructionDevice(*channel.send_done); in VerifyChannelInstructions()468 auto recv_device = GetInstructionDevice(*channel.recv); in VerifyChannelInstructions()469 auto recv_done_device = GetInstructionDevice(*channel.recv_done); in VerifyChannelInstructions()523 auto from_device = GetInstructionDevice(*channel.send); in DumpCollectedStats()524 auto to_device = GetInstructionDevice(*channel.recv); in DumpCollectedStats()546 auto device = GetInstructionDevice(*instruction); in DumpCollectedStats()
171 absl::optional<int64> GetInstructionDevice(