Home
last modified time | relevance | path

Searched defs:old_operand (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dwhile_loop_invariant_code_motion.cc67 auto get_new_operand = [&](HloInstruction* old_operand) { in CreateLoopInvariantCopy()
H A Dwhile_loop_expensive_invariant_code_motion.cc81 auto get_new_operand = [&](HloInstruction* old_operand) { in CreateLoopInvariantCopy()
H A Dhlo_module.cc642 HloInstruction* old_operand = in OutlineExpressionFromComputation() local
H A Dhlo_instruction.cc2689 auto old_operand = operand(operand_num); in ReplaceOperandWith() local
2701 HloInstruction* old_operand = mutable_operand(operand_num); in ReplaceOperandWithDifferentShape() local
H A Dspace_to_batch_converter.cc356 ConvolutionVisitor::GetSpatialDimsToSplit(HloInstruction* old_operand) { in GetSpatialDimsToSplit()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc3869 auto old_operand = input_tuple->mutable_operand(non_windowed_operand_index); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local