Searched refs:computation_id (Results 1 – 4 of 4) sorted by relevance
42 int64 computation_id = id_sequence.first; in CreateFromProto() local44 auto comp_it = id_to_computation.find(computation_id); in CreateFromProto()46 << "No computation exists in HLO module with id " << computation_id; in CreateFromProto()72 int64 computation_id = id_sequence.first; in ToProto() local75 (*proto.mutable_sequences())[computation_id]; in ToProto()
314 int64 computation_id = computation_proto.id(); in CreateFromProto() local315 TF_RET_CHECK(computation_id != -1); in CreateFromProto()316 TF_RET_CHECK(!ContainsKey(computation_map, computation_id)); in CreateFromProto()317 computation_map[computation_id] = computation.get(); in CreateFromProto()318 to_proto_id[computation.get()] = computation_id; in CreateFromProto()319 if (computation_id == proto.entry_computation_id()) { in CreateFromProto()
117 [&computation_map](int64 computation_id) { in CreateFromProto() argument118 return computation_map.at(computation_id); in CreateFromProto()598 for (const int64 computation_id : proto.called_computation_ids()) { in CreateFromProto() local600 computation_map.at(computation_id)); in CreateFromProto()
2758 int64 computation_id = GetNextId(); in AddCalledComputation() local2759 remapped_ids[new_computation.id()] = computation_id; in AddCalledComputation()2762 kNameSeparator, computation_id); in AddCalledComputation()2793 int64 computation_id = imported_computation.id(); in AddCalledComputation() local2794 embedded_.insert({computation_id, std::move(imported_computation)}); in AddCalledComputation()