Searched refs:BroadcastZeros (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_creation_utils_test.cc | 210 BroadcastZeros(module->entry_computation(), S32, {2, 2}); in TEST_F() 229 BroadcastZeros(module->entry_computation(), F32, {2, 2}); in TEST_F()
|
D | hlo_creation_utils.h | 283 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 | 549 HloInstruction* BroadcastZeros(HloComputation* computation, in BroadcastZeros() function
|
D | algebraic_simplifier.cc | 2737 BroadcastZeros(computation_, multiply->shape().element_type(), in HandleMultiply() 3701 HloInstruction* zero_like_a = BroadcastZeros( in TryRemainderToAnd()
|