Searched refs:MakeBroadcastHlo (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 215 return MakeBroadcastHlo(valid_index_reduced, {}, window_sizes); in CheckIndexValidity()
233 MakeBroadcastHlo(induction_var, /*broadcast_dimensions=*/{}, in ScatterLoopBody()
|
| D | hlo_creation_utils.h | 91 HloInstruction* MakeBroadcastHlo(HloInstruction* operand,
|
| D | hlo_creation_utils.cc | 176 HloInstruction* MakeBroadcastHlo(HloInstruction* operand, in MakeBroadcastHlo() function
442 return MakeBroadcastHlo(zero, /*broadcast_dimensions=*/{}, in BroadcastZeros()
|
| D | gather_expander.cc | 165 MakeBroadcastHlo(induction_var, /*broadcast_dimensions=*/{}, in GatherLoopBody()
|