Searched refs:reducer_id (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 102 HloComputationProto CopyReducer(int64 reducer_id, in CopyReducer() argument
106 SetProtoIdAndName(&reducer, StrCat("reduce_or"), kNameSeparator, reducer_id); in CopyReducer()
3467 int64 reducer_id = (*global_id)++; in BuildDynamicInferenceGraph() local
3469 new_instr->add_called_computation_ids(reducer_id); in BuildDynamicInferenceGraph()
3471 reducer_id, reducer, /*rewrite_into_pred=*/false, global_id)); in BuildDynamicInferenceGraph()
3549 int64 reducer_id = (*global_id)++; in BuildDynamicInferenceGraph() local
3551 new_instr->add_called_computation_ids(reducer_id); in BuildDynamicInferenceGraph()
3553 reducer_id, reducer, /*rewrite_into_pred=*/true, global_id)); in BuildDynamicInferenceGraph()
|