Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbroadcast_canonicalizer.cc54 auto new_broadcast = MakeBroadcastHlo(hlo->mutable_operand(0), new_dims, in Run() local
H A Dalgebraic_simplifier.cc603 HloInstruction* new_broadcast = target->AddInstruction( in ScalarMultiplyReduction() local
3523 HloInstruction* new_broadcast = user->AddInstruction( in HandleBroadcast() local
4048 HloInstruction* new_broadcast = in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand() local
4692 auto new_broadcast = in HandleSlice() local
4843 auto new_broadcast = in HandleDynamicSlice() local
H A Dspace_to_batch_converter.cc1450 auto new_broadcast = MakeBroadcastHlo(consumer->mutable_operand(0), in PropagateOnBroadcast() local
1777 HloInstruction* new_broadcast = nullptr; in Propagate() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
H A Dtf_jitrt_symbolic_shape_optimization.cc156 auto new_broadcast = simplifyBroadcast( in matchAndRewrite() local