Home
last modified time | relevance | path

Searched defs:zero_s32 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc98 auto zero_s32 = in PadEachPartitionWithHaloExchange() local
Dconvolution_handler.cc352 auto zero_s32 = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
Dspmd_partitioner_util.cc481 auto zero_s32 = b->AddInstruction( in TileToPartialReplicateHaloExchange() local
593 auto zero_s32 = b->AddInstruction( in PadFromPartialReplicateShape() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc749 HloInstruction* zero_s32 = comp->AddInstruction( in RewriteInputWithDynamicPadding() local