Searched refs:new_operand (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_inliner.cc | 47 TF_ASSIGN_OR_RETURN(HloInstruction * new_operand, Resolve(operand)); in DefaultAction()
48 new_operands.push_back(new_operand); in DefaultAction()
|
| D | hlo_instruction.cc | 1338 for (const HloInstruction* new_operand : new_operands) { in CloneWithNewOperands() local
1339 VLOG(3) << " %" << new_operand->name(); in CloneWithNewOperands()
1918 HloInstruction* new_operand) { in ReplaceOperandWith() argument
1921 new_operand->shape())) in ReplaceOperandWith()
1923 << new_operand->shape(); in ReplaceOperandWith()
1924 return ReplaceOperandWithDifferentShape(operand_num, new_operand); in ReplaceOperandWith()
1928 int64 operand_num, HloInstruction* new_operand) { in ReplaceOperandWithDifferentShape() argument
1932 if (old_operand == new_operand) { in ReplaceOperandWithDifferentShape()
1936 operands_[operand_num] = new_operand; in ReplaceOperandWithDifferentShape()
1939 << new_operand->name() << ", was " << old_operand->name(); in ReplaceOperandWithDifferentShape()
[all …]
|
| D | hlo_computation.cc | 988 HloInstruction* new_operand = replace(operand); in CloneWithReplacements() local
989 if (new_operand) { in CloneWithReplacements()
990 dfs_stack.emplace_back(new_operand); in CloneWithReplacements()
|
| D | hlo_instruction.h | 975 Status ReplaceOperandWith(int64 operand_num, HloInstruction* new_operand);
979 HloInstruction* new_operand);
1435 HloInstruction* AddFusionOperand(HloInstruction* new_operand);
|
| D | hlo_instructions.cc | 1215 HloInstruction* new_operand) { in AddFusionOperand() argument
1225 HloInstruction::CreateParameter(param_no, new_operand->shape(), in AddFusionOperand()
1227 AppendOperand(new_operand); in AddFusionOperand()
1301 auto new_operand = in MergeFusionInstructionIntoMultiOutput() local
1303 TF_CHECK_OK(unfused_instruction->ReplaceOperandWith(index, new_operand)); in MergeFusionInstructionIntoMultiOutput()
|
| D | hlo_instructions.h | 719 HloInstruction* AddFusionOperand(HloInstruction* new_operand);
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1762 for (XlaOp& new_operand : new_operands) { in Map()
1763 TF_ASSIGN_OR_RETURN(Shape shape, GetShape(new_operand)); in Map()
1766 TF_ASSIGN_OR_RETURN(new_operand, in Map()
1767 InDimBroadcast(output_shape, new_operand, {})); in Map()
1768 TF_ASSIGN_OR_RETURN(shape, GetShape(new_operand)); in Map()
1771 TF_ASSIGN_OR_RETURN(new_operand, in Map()
1772 AddBroadcastSequence(output_shape, new_operand)); in Map()
|