Home
last modified time | relevance | path

Searched defs:hlo (Results 1 – 25 of 75) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor.h80 virtual Status HandleMaximum(HloInstructionPtr hlo) { in HandleMaximum()
83 virtual Status HandleMinimum(HloInstructionPtr hlo) { in HandleMinimum()
87 virtual Status HandleConvert(HloInstructionPtr hlo) { in HandleConvert()
90 virtual Status HandleBitcastConvert(HloInstructionPtr hlo) { in HandleBitcastConvert()
93 virtual Status HandleCopy(HloInstructionPtr hlo) { in HandleCopy()
96 virtual Status HandleComplex(HloInstructionPtr hlo) { in HandleComplex()
99 virtual Status HandleMultiply(HloInstructionPtr hlo) { in HandleMultiply()
103 virtual Status HandlePower(HloInstructionPtr hlo) { in HandlePower()
106 virtual Status HandleSqrt(HloInstructionPtr hlo) { in HandleSqrt()
109 virtual Status HandleRsqrt(HloInstructionPtr hlo) { in HandleRsqrt()
[all …]
Ddynamic_dimension_inference.cc98 Status DynamicDimensionInferenceVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction()
110 HloInstruction* hlo) { in HandleGetTupleElement()
123 Status DynamicDimensionInferenceVisitor::HandleTuple(HloInstruction* hlo) { in HandleTuple()
133 Status DynamicDimensionInferenceVisitor::HandleBroadcast(HloInstruction* hlo) { in HandleBroadcast()
143 Status DynamicDimensionInferenceVisitor::HandlePad(HloInstruction* hlo) { in HandlePad()
166 Status DynamicDimensionInferenceVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce()
200 Status DynamicDimensionInferenceVisitor::HandleDot(HloInstruction* hlo) { in HandleDot()
251 Status DynamicDimensionInferenceVisitor::HandleTranspose(HloInstruction* hlo) { in HandleTranspose()
262 HloInstruction* hlo) { in HandleConvolution()
307 HloInstruction* hlo) { in PassThroughDynamicDimension()
[all …]
Dbfloat16_support.cc23 bool BFloat16Support::SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand()
79 const HloInstruction& hlo, int64 operand_index) { in EffectiveOperandPrecisionIsOutputPrecision()
133 const HloInstruction& hlo, int64 operand_index) const { in EffectiveOperandPrecisionIsBF16() argument
Ddfs_hlo_visitor_with_default_test.cc40 Status DefaultAction(HloInstruction* hlo) override { in TEST_F()
50 Status HandleElementwiseBinary(HloInstruction* hlo) override { in TEST_F()
56 Status HandleElementwiseUnary(HloInstruction* hlo) override { in TEST_F()
Dbfloat16_normalization.h75 bool SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand()
80 bool SupportsBF16Output(const HloInstruction& hlo) const override { in SupportsBF16Output()
84 bool SupportsMixedPrecisions(const HloInstruction& hlo) const override { in SupportsMixedPrecisions()
Dbfloat16_conversion_folding.cc68 HloInstruction* hlo) { in FoldOutputConversions()
80 HloInstruction* hlo, int64 operand_index) { in FoldOperandConversion()
93 bool AllUsersAreF32ToBF16Converts(const HloInstruction* hlo) { in AllUsersAreF32ToBF16Converts()
110 HloInstruction* hlo) { in TryFoldBF16Conversions()
152 Status BFloat16ConversionFoldingVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction()
Ddfs_hlo_visitor_with_default.h54 Status HandleElementwiseUnary(HloInstructionPtr hlo) override { in HandleElementwiseUnary()
57 Status HandleElementwiseBinary(HloInstructionPtr hlo) override { in HandleElementwiseBinary()
61 Status HandleBatchNormTraining(HloInstructionPtr hlo) override { in HandleBatchNormTraining()
65 Status HandleBatchNormInference(HloInstructionPtr hlo) override { in HandleBatchNormInference()
69 Status HandleBatchNormGrad(HloInstructionPtr hlo) override { in HandleBatchNormGrad()
94 Status HandleTriangularSolve(HloInstructionPtr hlo) override { in HandleTriangularSolve()
97 Status HandleCholesky(HloInstructionPtr hlo) override { in HandleCholesky()
103 Status HandleAllToAll(HloInstructionPtr hlo) override { in HandleAllToAll()
106 Status HandleCollectivePermute(HloInstructionPtr hlo) override { in HandleCollectivePermute()
109 Status HandleReplicaId(HloInstructionPtr hlo) override { in HandleReplicaId()
Dbfloat16_normalization.cc80 HloInstruction* hlo, PrimitiveType to, HloComputation* computation) { in InsertConvertAfterOutput()
103 HloInstruction* hlo, PrimitiveType to, HloComputation* computation) { in ChangeOutputTypeThenInsertConvertBack()
110 HloInstruction* hlo, int64 operand_idx, PrimitiveType to, in InsertConvertBeforeOperand()
121 HloInstruction* hlo, absl::Span<HloComputation* const> bf16_called_comps) { in ConvertCalledComputations()
153 HloInstruction* hlo) { in HandleMultipleOutputs()
269 Status BFloat16NormalizationVisitor::HandleInstruction(HloInstruction* hlo) { in HandleInstruction()
374 Status BFloat16NormalizationVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction()
Dhlo_element_type_converter.cc38 HloInstruction* ToElementType(HloInstruction* hlo, PrimitiveType type) { in ToElementType()
48 bool HasOperandType(HloInstruction* hlo, PrimitiveType type) { in HasOperandType()
85 HloInstruction* ConvertTupleElements(HloInstruction* hlo, in ConvertTupleElements()
125 for (auto* hlo : computation->MakeInstructionPostOrder()) { in Run() local
Dbfloat16_propagation.cc206 bool BFloat16Propagation::AllUsersConsumeBF16(const HloInstruction& hlo, in AllUsersConsumeBF16()
311 void BFloat16Propagation::DetermineInstructionPrecision(HloInstruction* hlo, in DetermineInstructionPrecision()
389 HloInstruction* hlo) { in InstructionIsCandidateForBF16Output()
407 HloInstruction* hlo) { in AdjustCalledComputationParameters()
451 void BFloat16Propagation::AdjustCalledComputationRoot(HloInstruction* hlo) { in AdjustCalledComputationRoot()
509 auto hlo = *inst_it; in ResolveInconsistencyOfAliasingBuffersHelper() local
629 auto hlo = *inst_it; in ResolveInconsistentFusions() local
673 for (auto hlo : computation->MakeInstructionPostOrder()) { in ResolveConvertedConstants() local
691 for (auto hlo : computation->MakeInstructionPostOrder()) { in SkipNoopConversions() local
829 HloInstruction* hlo, const ShapeIndex& index) const { in OutputTypeAfterChange()
[all …]
Ddfs_hlo_visitor.cc29 HloInstructionPtr hlo) { in HandleElementwiseUnary()
36 HloInstructionPtr hlo) { in HandleElementwiseBinary()
Dhlo_reachability.cc26 for (const HloInstruction* hlo : instructions) { in HloReachabilityMap() local
84 for (const HloInstruction* hlo : all) { in Build() local
96 const auto add_dependencies = [&add_input](const HloInstruction* hlo) { in Build()
Dcompiler.cc32 Compiler::ComputeBackendConfigs(const HloInstruction& hlo, in ComputeBackendConfigs()
39 Compiler::ComputeDefaultBackendConfig(const HloInstruction& hlo, in ComputeDefaultBackendConfig()
Dhlo_pass_pipeline.cc36 HloT* hlo, absl::string_view after_pass_name) { in RunInvariantCheckers()
56 HloT* hlo, absl::Span<HloPassInterface* const> passes) { in RunPassesInternal()
Dhlo_cost_analysis.cc42 Status HloCostAnalysis::Preprocess(const HloInstruction* hlo) { in Preprocess()
61 Status HloCostAnalysis::Postprocess(const HloInstruction* hlo) { in Postprocess()
115 const HloInstruction& hlo, const string& key, in GetPropertyForHlo()
132 Status HloCostAnalysis::HandleElementwiseUnary(const HloInstruction* hlo) { in HandleElementwiseUnary()
136 Status HloCostAnalysis::HandleElementwiseBinary(const HloInstruction* hlo) { in HandleElementwiseBinary()
148 Status HloCostAnalysis::HandleReducePrecision(const HloInstruction* hlo) { in HandleReducePrecision()
179 Status HloCostAnalysis::HandleSelect(const HloInstruction* hlo) { in HandleSelect()
550 Status HloCostAnalysis::HandleTriangularSolve(const HloInstruction* hlo) { in HandleTriangularSolve()
564 Status HloCostAnalysis::HandleCholesky(const HloInstruction* hlo) { in HandleCholesky()
593 Status HloCostAnalysis::HandleAllToAll(const HloInstruction* hlo) { in HandleAllToAll()
Dhlo_execution_profile.cc94 for (const HloInstruction* hlo : computation->instructions()) { in CreateHloProfilePrinterData() local
133 void HloExecutionProfile::SetCyclesTakenBy(const HloInstruction* hlo, in SetCyclesTakenBy()
Dhlo_verifier.cc42 bool IsCallerInstruction(HloInstruction* hlo) { in IsCallerInstruction()
63 Status CheckOperandCount(const HloInstruction* hlo, int expected) { in CheckOperandCount()
85 Status ShapeVerifier::Preprocess(HloInstruction* hlo) { in Preprocess()
100 Status ShapeVerifier::HandleElementwiseUnary(HloInstruction* hlo) { in HandleElementwiseUnary()
104 Status ShapeVerifier::HandleElementwiseBinary(HloInstruction* hlo) { in HandleElementwiseBinary()
172 Status ShapeVerifier::HandleTriangularSolve(HloInstruction* hlo) { in HandleTriangularSolve()
180 Status ShapeVerifier::HandleCholesky(HloInstruction* hlo) { in HandleCholesky()
195 Status ShapeVerifier::HandleAllToAll(HloInstruction* hlo) { in HandleAllToAll()
204 Status ShapeVerifier::HandleReplicaId(HloInstruction* hlo) { in HandleReplicaId()
208 Status ShapeVerifier::HandleCollectivePermute(HloInstruction* hlo) { in HandleCollectivePermute()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmemset_thunk.h36 const HloInstruction* hlo) in MemzeroThunk()
53 const HloInstruction* hlo) in Memset32BitValueThunk()
Dir_emission_utils.cc86 bool ImplementedAsGemm(const HloInstruction& hlo) { in ImplementedAsGemm()
116 bool IsCustomCallToDnnBatchNorm(const HloInstruction& hlo) { in IsCustomCallToDnnBatchNorm()
134 bool IsCustomCallToDnnConvolution(const HloInstruction& hlo) { in IsCustomCallToDnnConvolution()
147 bool IsCustomCallToCusolver(const HloInstruction& hlo) { in IsCustomCallToCusolver()
155 bool ImplementedAsLibraryCall(const HloInstruction& hlo) { in ImplementedAsLibraryCall()
Dstream_assignment.cc36 void StreamAssignment::AssignStreamToHlo(const HloInstruction* hlo, in AssignStreamToHlo()
66 const HloInstruction& hlo, const StreamAssignment& stream_assignment, in ComputeStreamToAssign()
135 for (const auto* hlo : computation.MakeInstructionPostOrder()) { in AssignStreams() local
Dhlo_to_ir_bindings.cc153 llvm::Value* HloToIrBindings::GetTypedIrValue(const HloInstruction& hlo, in GetTypedIrValue()
176 void HloToIrBindings::BindHloToIrValue(const HloInstruction& hlo, in BindHloToIrValue()
194 const HloInstruction& hlo, const HloInstruction& consumer, in BuffersInvariantWithinConsumer()
213 llvm_ir::IrArray HloToIrBindings::GetIrArray(const HloInstruction& hlo, in GetIrArray()
Dir_emitter_nested.cc102 for (const auto* hlo : nested_computation.instructions()) { in EmitBasePointersForNestedComputation() local
117 const HloInstruction& hlo, in EmitTargetElementLoop()
Dgpu_hlo_schedule.cc86 for (const HloInstruction* hlo : thunk_launch_order) { in GpuHloOrdering() local
163 for (auto* hlo : computation->instructions()) { in BFSLaunchOrder() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.cc30 bool CanBeLoopFused(const HloInstruction& hlo) { in CanBeLoopFused()
46 bool IsNonComplexMatrixVectorDot(const HloInstruction* hlo) { in IsNonComplexMatrixVectorDot()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dalias_analysis.cc33 void AliasAnalysis::AddAliasingInformationToIrArray(const HloInstruction& hlo, in AddAliasingInformationToIrArray()
125 const BufferAssignment& assignment, const HloInstruction& hlo) { in GetNoaliasMetadataForBuffer()

123