Searched refs:partitioned_instructions_ (Results 1 – 1 of 1) sorted by relevance
456 CHECK_EQ(partitioned_instructions_.count(hlo), 1); in GetPartitionedHlo()457 return partitioned_instructions_.find(hlo)->second; in GetPartitionedHlo()463 CHECK_EQ(partitioned_instructions_.count(hlo), 0); in SetPartitionedHlo()464 partitioned_instructions_.emplace(hlo, partitioned_hlo); in SetPartitionedHlo()539 ConstHloInstructionMap<PartitionedHlo> partitioned_instructions_; variable