Home
last modified time | relevance | path

Searched refs:fused_parameter (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion.cc173 auto fused_parameter = consumer->fused_parameter(producer_operand_index); in ShouldFuseInexpensiveChecks() local
175 fused_parameter->users(); in ShouldFuseInexpensiveChecks()
Dir_emitter_unnested.cc3352 const HloInstruction* input = fusion->fused_parameter(input_ids[i]); in FilterInputsForShmemTranspose()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation.cc137 auto* fused_parameter = inst->fused_parameter(i); in RevertIfFusionInternalBF16Changes() local
138 if (has_changes(fused_parameter)) { in RevertIfFusionInternalBF16Changes()
139 changes_to_bf16_.erase(fused_parameter); in RevertIfFusionInternalBF16Changes()
248 auto* fused_parameter = in AllUsersConsumeBF16() local
249 use.instruction->fused_parameter(use.operand_number); in AllUsersConsumeBF16()
250 if (OutputTypeAfterChange(fused_parameter, use.operand_index) != BF16) { in AllUsersConsumeBF16()
Dhlo_instructions.cc1192 worklist.push_back(fused_parameter(operand_idx.value())); in IsElementwiseImpl()
1193 visited.insert(fused_parameter(operand_idx.value())); in IsElementwiseImpl()
1223 HloInstruction* fused_parameter = in AddFusionOperand() local
1228 return fused_parameter; in AddFusionOperand()
1340 HloInstruction* HloFusionInstruction::fused_parameter( in fused_parameter() function in xla::HloFusionInstruction
1418 HloInstruction* fused_parameter = fused_parameters[operand_num]; in CloneAndFuseInternal() local
1419 TF_CHECK_OK(fused_parameter->ReplaceAllUsesWith(clone)); in CloneAndFuseInternal()
1585 TF_RETURN_IF_ERROR(fused_parameter(i)->ReplaceAllUsesWith( in DeduplicateFusionOperands()
1586 fused_parameter(emplace_result.first->second))); in DeduplicateFusionOperands()
Dhlo_dataflow_analysis.cc935 user->fused_parameter(use.operand_number); in DoesNotUseOperandBuffer()
1038 user->fused_parameter(user->operand_index(operand)); in CanShareOperandBufferWithUser()
Dlayout_assignment_test.cc199 layout, fusion->fused_parameter(0)->shape().layout())); in TEST_F()
201 layout, fusion->fused_parameter(1)->shape().layout())); in TEST_F()
Dtuple_points_to_analysis.cc732 user->fused_parameter(user->operand_index(operand)); in CanShareOperandBufferWithUser()
Dhlo_instruction.h1470 HloInstruction* fused_parameter(int64 parameter_number) const;
Dhlo_instructions.h778 HloInstruction* fused_parameter(int64 parameter_number) const;
Dhlo_instruction.cc3342 HloInstruction* HloInstruction::fused_parameter(int64 parameter_number) const { in fused_parameter() function in xla::HloInstruction
3343 return Cast<HloFusionInstruction>(this)->fused_parameter(parameter_number); in fused_parameter()
Dlayout_assignment.cc737 fusion->fused_parameter(i)->shape(), fusion->operand(i)->shape())); in CheckFusionLayout()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.cc277 auto instruction = consumer->fused_parameter(operand_num); in IsFusedIrEmitterInefficient()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc410 instruction->fused_parameter(op_num); in FindConstrainedUses()