Searched refs:called_computations_ (Results 1 – 3 of 3) sorted by relevance
65 called_computations_(called_computations), in CallSite()73 return called_computations_; in called_computations()86 const std::vector<HloComputation*> called_computations_; variable
1204 return called_computations_; in called_computations()1212 for (int64 i = 0; i < called_computations_.size(); ++i) { in ReplaceCalledComputations()1213 called_computations_[i] = map_function(called_computations_[i]); in ReplaceCalledComputations()1225 void ClearCalledComputations() { called_computations_.clear(); } in ClearCalledComputations()1642 called_computations_.push_back(computation); in AppendComputation()1648 called_computations_[index] = computation; in set_called_computation()1772 std::vector<HloComputation*> called_computations_; variable
599 instruction->called_computations_.push_back( in CreateFromProto()947 instruction->called_computations_.push_back(body); in CreateWhile()948 instruction->called_computations_.push_back(condition); in CreateWhile()964 instruction->called_computations_.push_back(true_computation); in CreateConditional()965 instruction->called_computations_.push_back(false_computation); in CreateConditional()978 instruction->called_computations_.push_back(branch_computations[i]); in CreateConditional()1276 instruction->called_computations_.push_back(computation); in CreateCall()1997 CHECK_EQ(called_computations_.size(), 1); in to_apply()1998 return called_computations_[0]; in to_apply()2017 CHECK_EQ(called_computations_.size(), 1); in set_to_apply()[all …]