Searched refs:AddCalledComputation (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 626 AddCalledComputation(computation, &instr); in Call()
1702 AddCalledComputation(comparator, &instr); in Sort()
1760 AddCalledComputation(computation, &instr); in Map()
1834 AddCalledComputation(body, &instr); in While()
1835 AddCalledComputation(condition, &instr); in While()
1885 AddCalledComputation(update_computation, &instr); in Scatter()
1926 AddCalledComputation(*branch_computation, &instr); in Conditional()
1977 AddCalledComputation(computation, &instr); in Reduce()
2039 AddCalledComputation(computation, &instr); in ReduceWindowWithGeneralPadding()
2153 AddCalledComputation(computation, &instr); in CrossReplicaSum()
[all …]
|
| D | xla_builder.h | 588 void AddCalledComputation(const XlaComputation& computation,
|