Searched refs:branch_index_table (Results 1 – 1 of 1) sorted by relevance
2556 auto branch_index_table = b_.AddInstruction(HloInstruction::CreateConstant( in HandleInfeed() local2559 ShapeUtil::MakeShape(S32, {1}), branch_index_table, {partition_id_}, in HandleInfeed()2959 auto branch_index_table = b_.AddInstruction(HloInstruction::CreateConstant( in HandleOutfeed() local2962 ShapeUtil::MakeShape(S32, {1}), branch_index_table, {partition_id_}, in HandleOutfeed()