Home
last modified time | relevance | path

Searched refs:BroadcastZeros (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils_test.cc195 BroadcastZeros(module->entry_computation(), S32, {2, 2}); in TEST_F()
214 BroadcastZeros(module->entry_computation(), F32, {2, 2}); in TEST_F()
Dhlo_creation_utils.h216 HloInstruction* BroadcastZeros(HloComputation* computation,
Dgather_expander.cc244 return BroadcastZeros(computation, element_type, in CreateGatherLoopAccumulatorInitValue()
Dscatter_expander.cc182 BroadcastZeros(computation, index->shape().element_type(), in CheckIndexValidity()
Dhlo_creation_utils.cc437 HloInstruction* BroadcastZeros(HloComputation* computation, in BroadcastZeros() function
Dalgebraic_simplifier.cc872 HloInstruction* zero_like_a = BroadcastZeros( in TryDivideToShift()
2474 HloInstruction* zero_like_a = BroadcastZeros( in TryRemainderToAnd()