| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_inliner.cc | 46 TF_ASSIGN_OR_RETURN(HloInstruction * new_operand, Resolve(operand)); in DefaultAction()
47 new_operands.push_back(new_operand); in DefaultAction()
|
| D | bfloat16_normalization.cc | 222 TF_ASSIGN_OR_RETURN(auto new_operand, in InsertConvertBeforeOperand()
224 if (new_operand == operand) { in InsertConvertBeforeOperand()
228 hlo->ReplaceOperandWithDifferentShape(operand_idx, new_operand)); in InsertConvertBeforeOperand()
|
| D | hlo_instruction.cc | 1649 for (const HloInstruction* new_operand : new_operands) { in CloneWithNewOperands() local
1650 VLOG(3) << " %" << new_operand->name(); in CloneWithNewOperands()
2330 HloInstruction* new_operand) { in ReplaceOperandWith() argument
2333 new_operand->shape())) in ReplaceOperandWith()
2335 << new_operand->shape(); in ReplaceOperandWith()
2336 return ReplaceOperandWithDifferentShape(operand_num, new_operand); in ReplaceOperandWith()
2340 int64 operand_num, HloInstruction* new_operand) { in ReplaceOperandWithDifferentShape() argument
2344 if (old_operand == new_operand) { in ReplaceOperandWithDifferentShape()
2348 operands_[operand_num] = new_operand; in ReplaceOperandWithDifferentShape()
2351 << new_operand->name() << ", was " << old_operand->name(); in ReplaceOperandWithDifferentShape()
[all …]
|
| D | hlo_computation.cc | 1085 const HloInstruction* new_operand = replace(operand); in CloneWithReplacements() local
1086 if (new_operand) { in CloneWithReplacements()
1087 dfs_stack.emplace_back(new_operand); in CloneWithReplacements()
|
| D | algebraic_simplifier.cc | 1033 TF_ASSIGN_OR_RETURN(HloInstruction * new_operand, in HandleAdd()
1045 add->shape(), new_operand, new_index, new_update, in HandleAdd()
1048 TF_ASSIGN_OR_RETURN(HloInstruction * new_operand, in HandleAdd()
1051 TF_RETURN_IF_ERROR(rhs->ReplaceOperandWith(0, new_operand)); in HandleAdd()
1056 HloInstruction * new_operand, in HandleAdd()
1058 TF_RETURN_IF_ERROR(rhs->ReplaceOperandWith(0, new_operand)); in HandleAdd()
1062 HloInstruction * new_operand, in HandleAdd()
1064 TF_RETURN_IF_ERROR(lhs->ReplaceOperandWith(0, new_operand)); in HandleAdd()
2517 HloInstruction* new_operand = gather->mutable_operand(0); in HandleGather() local
2519 TF_ASSIGN_OR_RETURN(new_operand, in HandleGather()
[all …]
|
| D | dynamic_dimension_inference.cc | 1309 HloInstruction* new_operand = hlo->mutable_operand(operand_index); in HandleConditional() local
1313 new_operand = TupleUtil::AppendSuffix(original_input, operands_to_add); in HandleConditional()
1316 WidenComputation(branch_computation, new_operand->shape())); in HandleConditional()
1366 new_operands.push_back(new_operand); in HandleConditional()
|
| D | hlo_instruction.h | 1197 Status ReplaceOperandWith(int64 operand_num, HloInstruction* new_operand);
1201 HloInstruction* new_operand);
1715 HloInstruction* AddFusionOperand(HloInstruction* new_operand);
|
| D | hlo_instructions.cc | 1448 HloInstruction* new_operand) { in AddFusionOperand() argument
1455 HloInstruction::CreateParameter(param_no, new_operand->shape(), in AddFusionOperand()
1457 AppendOperand(new_operand); in AddFusionOperand()
1558 auto new_operand = in MergeFusionInstructionIntoMultiOutput() local
1560 TF_CHECK_OK(unfused_instruction->ReplaceOperandWith(index, new_operand)); in MergeFusionInstructionIntoMultiOutput()
|
| D | space_to_batch_converter.cc | 1182 auto new_operand = old_to_new_instrs_[first_operand]; in SupportedOpForPropagation() local
1183 auto permute_dims = instr_to_dim_permute_map_[new_operand]; in SupportedOpForPropagation()
1188 if (new_operand->shape().dimensions(new_space_dim) % in SupportedOpForPropagation()
|
| D | hlo_instructions.h | 903 HloInstruction* AddFusionOperand(HloInstruction* new_operand);
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2147 for (XlaOp& new_operand : new_operands) { in Map()
2148 TF_ASSIGN_OR_RETURN(const Shape* shape, GetShapePtr(new_operand)); in Map()
2151 TF_ASSIGN_OR_RETURN(new_operand, in Map()
2152 InDimBroadcast(output_shape, new_operand, {})); in Map()
2153 TF_ASSIGN_OR_RETURN(shape, GetShapePtr(new_operand)); in Map()
2156 TF_ASSIGN_OR_RETURN(new_operand, in Map()
2157 AddBroadcastSequence(output_shape, new_operand)); in Map()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 3118 for (auto new_operand : find_result.second) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
3119 auto res = new_operands_set.insert(new_operand); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3121 new_operands.push_back(new_operand); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
|