Searched refs:BroadcastZeros (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_creation_utils_test.cc | 195 BroadcastZeros(module->entry_computation(), S32, {2, 2}); in TEST_F() 214 BroadcastZeros(module->entry_computation(), F32, {2, 2}); in TEST_F()
|
D | hlo_creation_utils.h | 216 HloInstruction* BroadcastZeros(HloComputation* computation,
|
D | gather_expander.cc | 244 return BroadcastZeros(computation, element_type, in CreateGatherLoopAccumulatorInitValue()
|
D | scatter_expander.cc | 182 BroadcastZeros(computation, index->shape().element_type(), in CheckIndexValidity()
|
D | hlo_creation_utils.cc | 437 HloInstruction* BroadcastZeros(HloComputation* computation, in BroadcastZeros() function
|
D | algebraic_simplifier.cc | 872 HloInstruction* zero_like_a = BroadcastZeros( in TryDivideToShift() 2474 HloInstruction* zero_like_a = BroadcastZeros( in TryRemainderToAnd()
|