Home
last modified time | relevance | path

Searched refs:ReplaceOperandWithDifferentShape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvert_operand_folding.cc53 TF_RETURN_IF_ERROR(instruction->ReplaceOperandWithDifferentShape( in ExpandInstruction()
Doperand_upcaster.cc71 instruction->ReplaceOperandWithDifferentShape(i, convert_inst)); in ExpandInstruction()
Dspace_to_batch_converter.cc1253 new_consumer->ReplaceOperandWithDifferentShape(i, new_broadcast)); in Propagate()
1314 new_consumer->ReplaceOperandWithDifferentShape(i, operand_to_use)); in Propagate()
1321 TF_CHECK_OK(new_consumer->ReplaceOperandWithDifferentShape( in Propagate()
1328 new_consumer->ReplaceOperandWithDifferentShape(i, new_constant)); in Propagate()
1385 new_consumer->ReplaceOperandWithDifferentShape(0, first_operand)); in Propagate()
1486 new_consumer->ReplaceOperandWithDifferentShape(0, first_operand)); in Propagate()
1489 new_consumer->ReplaceOperandWithDifferentShape(1, second_operand)); in Propagate()
1630 new_consumer->ReplaceOperandWithDifferentShape(0, first_operand)); in Propagate()
Dconditional_simplifier.cc278 conditional->ReplaceOperandWithDifferentShape(branch + 1, new_tuple)); in TryRemoveUnusedConditionalOperands()
Dbfloat16_normalization.cc228 hlo->ReplaceOperandWithDifferentShape(operand_idx, new_operand)); in InsertConvertBeforeOperand()
Dhlo_instruction.h1200 Status ReplaceOperandWithDifferentShape(int64 operand_num,
Dhlo_instruction.cc2336 return ReplaceOperandWithDifferentShape(operand_num, new_operand); in ReplaceOperandWith()
2339 Status HloInstruction::ReplaceOperandWithDifferentShape( in ReplaceOperandWithDifferentShape() function in xla::HloInstruction
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_rewriter.cc668 TF_RETURN_IF_ERROR(conv->ReplaceOperandWithDifferentShape( in CreateCustomCallHelper()
672 TF_RETURN_IF_ERROR(conv->ReplaceOperandWithDifferentShape( in CreateCustomCallHelper()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc2944 TF_RETURN_IF_ERROR(input_tuple->ReplaceOperandWithDifferentShape( in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
2962 TF_RETURN_IF_ERROR(body->root_instruction()->ReplaceOperandWithDifferentShape( in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
3177 input_tuple->ReplaceOperandWithDifferentShape(2, new_input_subtuple)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3371 body_root->ReplaceOperandWithDifferentShape(2, new_output_inside)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()