Searched refs:instruction_value_map_ (Results 1 – 2 of 2) sorted by relevance
180 instruction_value_map_[hlo_parameter] = arguments[i]; in ImportInstructionsImpl()187 instruction_value_map_[instruction] = new_operation->getResult(0); in ImportInstructionsImpl()865 auto input_it = instruction_value_map_.find(operand); in GetOperands()866 if (input_it == instruction_value_map_.end()) { in GetOperands()888 auto lookup = instruction_value_map_.find(instruction); in GetMlirValue()889 if (lookup != instruction_value_map_.end()) { in GetMlirValue()
166 std::unordered_map<xla::HloInstruction*, mlir::Value> instruction_value_map_; variable