Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_propagation.cc37 instruction->fused_parameter(operand_idx), in Run()
103 use.instruction->fused_parameter(use.operand_number), memory_space); in Propagate()
Dbfloat16_propagation.cc139 auto* fused_parameter = inst->fused_parameter(i); in RevertIfFusionInternalBF16Changes() local
140 if (has_changes(fused_parameter)) { in RevertIfFusionInternalBF16Changes()
141 changes_to_bf16_.erase(fused_parameter); in RevertIfFusionInternalBF16Changes()
277 auto* fused_parameter = in AllUsersConsumeBF16() local
278 use.instruction->fused_parameter(use.operand_number); in AllUsersConsumeBF16()
279 if (OutputTypeAfterChange(fused_parameter, use.operand_index) != BF16) { in AllUsersConsumeBF16()
Dfusion_node_indexing_evaluation.cc73 cnt += user->fused_parameter(operand_index)->user_count(); in UserCount()
Dhlo_instructions.cc1425 worklist.push_back(fused_parameter(operand_idx.value())); in IsElementwiseImpl()
1426 visited.insert(fused_parameter(operand_idx.value())); in IsElementwiseImpl()
1453 HloInstruction* fused_parameter = in AddFusionOperand() local
1458 return fused_parameter; in AddFusionOperand()
1609 HloInstruction* HloFusionInstruction::fused_parameter( in fused_parameter() function in xla::HloFusionInstruction
1687 HloInstruction* fused_parameter = fused_parameters[operand_num]; in CloneAndFuseInternal() local
1688 TF_CHECK_OK(fused_parameter->ReplaceAllUsesWith(clone)); in CloneAndFuseInternal()
1851 TF_RETURN_IF_ERROR(fused_parameter(i)->ReplaceAllUsesWith( in DeduplicateFusionOperands()
1852 fused_parameter(emplace_result.first->second))); in DeduplicateFusionOperands()
Dhlo_cost_analysis.cc155 size += FusionParameterReadBytes(user->fused_parameter(idx)); in FusionParameterReadBytes()
898 const HloInstruction* operand = fusion->fused_parameter(i); in HandleFusion()
Dhlo_dataflow_analysis.cc1204 user->fused_parameter(use.operand_number); in DoesNotUseOperandBuffer()
1302 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()
Dhlo_instruction.h1750 HloInstruction* fused_parameter(int64 parameter_number) const;
Dhlo_instructions.h962 HloInstruction* fused_parameter(int64 parameter_number) const;
Dhlo_instruction.cc4037 HloInstruction* HloInstruction::fused_parameter(int64 parameter_number) const { in fused_parameter() function in xla::HloInstruction
4038 return Cast<HloFusionInstruction>(this)->fused_parameter(parameter_number); in fused_parameter()
Dlayout_assignment.cc755 TF_RET_CHECK(LayoutsInShapesEqual(fusion->fused_parameter(i)->shape(), in CheckFusionLayout()
Dmemory_space_assignment.cc49 LooksLikeAnActivation(user->fused_parameter(i))) { in LooksLikeAnActivation()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc690 fusion->fused_parameter(i), in BindFusionArguments()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc519 instruction->fused_parameter(op_num); in FindConstrainedUses()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc3327 fusion->fused_parameter(i), in BindFusionArguments()