Searched refs:embedded_ (Results 1 – 2 of 2) sorted by relevance
501 for (auto& e : embedded_) { in Build()516 this->embedded_.clear(); in Build()3466 &embedded_[new_instr->called_computation_ids(0)]; in BuildDynamicInferenceGraph()3548 auto* reducer = &embedded_[new_instr->called_computation_ids(0)]; in BuildDynamicInferenceGraph()3870 for (auto& e : embedded_) { in BuildConstantSubGraph()4051 embedded_.insert({computation_id, std::move(imported_computation)}); in AddCalledComputation()
1028 std::map<int64, HloComputationProto> embedded_; variable