Searched refs:param_instructions_ (Results 1 – 2 of 2) sorted by relevance
75 param_instructions_.resize(parameter_count, nullptr); in HloComputation()83 CHECK(param_instructions_[param_no] == nullptr) in HloComputation()86 param_instructions_[param_no] = instruction.get(); in HloComputation()120 CHECK(fusion_instruction_->operand_count() == param_instructions_.size()); in AddParameter()122 param_instructions_.push_back(instruction.get()); in AddParameter()139 param_instructions_.push_back(instruction.get()); in AddEntryComputationParameter()147 CHECK_LT(param_no, param_instructions_.size()); in RemoveParameter()149 HloInstruction* param_instruction = param_instructions_[param_no]; in RemoveParameter()150 auto param_instruction_iterator = param_instructions_.begin() + param_no; in RemoveParameter()151 param_instructions_.erase(param_instruction_iterator); in RemoveParameter()[all …]
156 int64 num_parameters() const { return param_instructions_.size(); } in num_parameters()161 CHECK_LT(param_no, static_cast<int64>(param_instructions_.size())) in parameter_instruction()163 return param_instructions_[param_no]; in parameter_instruction()167 return param_instructions_; in parameter_instructions()473 std::vector<HloInstruction*> param_instructions_; variable