Home
last modified time | relevance | path

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

1234

/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 …]
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_support.cc23 bool BFloat16Support::SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand()
79 const HloInstruction& hlo, int64 operand_index) { in EffectiveOperandPrecisionIsOutputPrecision()
135 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()
Ddynamic_dimension_inference.cc128 Status DynamicDimensionInferenceVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction()
141 HloInstruction* hlo) { in HandleGetTupleElement()
156 Status DynamicDimensionInferenceVisitor::HandleTuple(HloInstruction* hlo) { in HandleTuple()
168 Status DynamicDimensionInferenceVisitor::HandleBroadcast(HloInstruction* hlo) { in HandleBroadcast() argument
180 Status DynamicDimensionInferenceVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall()
212 Status DynamicDimensionInferenceVisitor::HandleSort(HloInstruction* hlo) { in HandleSort()
230 Status DynamicDimensionInferenceVisitor::HandlePad(HloInstruction* hlo) { in HandlePad()
254 Status DynamicDimensionInferenceVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce()
304 Status DynamicDimensionInferenceVisitor::HandleDot(HloInstruction* hlo) { in HandleDot()
389 Status DynamicDimensionInferenceVisitor::HandleTranspose(HloInstruction* hlo) { in HandleTranspose()
[all …]
Dbfloat16_conversion_folding.cc75 HloInstruction* hlo) { in FoldOutputConversions()
88 HloInstruction* hlo, int64 operand_index) { in FoldOperandConversion()
101 bool AllUsersAreF32ToBF16Converts(const HloInstruction* hlo) { in AllUsersAreF32ToBF16Converts()
118 HloInstruction* hlo) { in TryFoldBF16Conversions()
160 Status BFloat16ConversionFoldingVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction()
Ddfs_hlo_visitor_with_default.h55 Status HandleElementwiseUnary(HloInstructionPtr hlo) override { in HandleElementwiseUnary()
58 Status HandleElementwiseBinary(HloInstructionPtr hlo) override { in HandleElementwiseBinary()
62 Status HandleBatchNormTraining(HloInstructionPtr hlo) override { in HandleBatchNormTraining()
66 Status HandleBatchNormInference(HloInstructionPtr hlo) override { in HandleBatchNormInference()
70 Status HandleBatchNormGrad(HloInstructionPtr hlo) override { in HandleBatchNormGrad()
95 Status HandleTriangularSolve(HloInstructionPtr hlo) override { in HandleTriangularSolve()
98 Status HandleCholesky(HloInstructionPtr hlo) override { in HandleCholesky()
104 Status HandleAllToAll(HloInstructionPtr hlo) override { in HandleAllToAll()
107 Status HandleCollectivePermute(HloInstructionPtr hlo) override { in HandleCollectivePermute()
110 Status HandleReplicaId(HloInstructionPtr hlo) override { in HandleReplicaId()
[all …]
Dbfloat16_normalization.cc118 HloInstruction* hlo, PrimitiveType from, PrimitiveType to, in ConvertType()
150 HloInstruction* hlo, PrimitiveType from, PrimitiveType to, in InsertConvertAfterOutput()
171 HloInstruction* hlo, PrimitiveType from, PrimitiveType to, in ChangeOutputTypeThenInsertConvertBack()
219 HloInstruction* hlo, int64 operand_idx, PrimitiveType from, in InsertConvertBeforeOperand()
234 HloInstruction* hlo, absl::Span<HloComputation* const> bf16_called_comps) { in ConvertCalledComputations()
262 HloInstruction* hlo) { in HandleMultipleOutputs()
383 Status BFloat16NormalizationVisitor::HandleInstruction(HloInstruction* hlo) { in HandleInstruction()
489 Status BFloat16NormalizationVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction()
517 Status BFloat16NormalizationVisitor::Preprocess(HloInstruction* hlo) { in Preprocess()
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()
333 void BFloat16Propagation::DetermineInstructionPrecision(HloInstruction* hlo, in DetermineInstructionPrecision()
405 HloInstruction* hlo) { in InstructionIsCandidateForBF16Output()
423 HloInstruction* hlo) { in AdjustCalledComputationParameters()
467 void BFloat16Propagation::AdjustCalledComputationRoot(HloInstruction* hlo) { in AdjustCalledComputationRoot()
525 auto hlo = *inst_it; in ResolveInconsistencyOfAliasingBuffersHelper() local
645 auto hlo = *inst_it; in ResolveInconsistentFusions() local
689 for (auto hlo : computation->MakeInstructionPostOrder()) { in ResolveConvertedConstants() local
709 for (auto hlo : computation->MakeInstructionPostOrder()) { in SkipNoopConversions() local
881 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
105 const auto add_dependencies = [&add_input](const HloInstruction* hlo) { in Build()
114 for (const HloInstruction* hlo : all) { in Build() local
Dhlo_query.cc126 for (auto hlo : computation->instructions()) { in ContainsLayoutConstrainedAllReduce() local
139 for (const HloInstruction* hlo : comp->instructions()) { in NextChannelId() local
153 for (auto hlo : computation->instructions()) { in HasX64TransformedHostTransfer() local
Dhlo_cost_analysis.cc47 Status HloCostAnalysis::Preprocess(const HloInstruction* hlo) { in Preprocess()
69 Status HloCostAnalysis::Postprocess(const HloInstruction* hlo) { in Postprocess()
125 const HloInstruction& hlo, const string& key, in GetPropertyForHlo()
184 Status HloCostAnalysis::HandleElementwiseUnary(const HloInstruction* hlo) { in HandleElementwiseUnary()
188 Status HloCostAnalysis::HandleElementwiseBinary(const HloInstruction* hlo) { in HandleElementwiseBinary()
200 Status HloCostAnalysis::HandleReducePrecision(const HloInstruction* hlo) { in HandleReducePrecision()
236 Status HloCostAnalysis::HandleSelect(const HloInstruction* hlo) { in HandleSelect()
672 Status HloCostAnalysis::HandleTriangularSolve(const HloInstruction* hlo) { in HandleTriangularSolve()
691 Status HloCostAnalysis::HandleCholesky(const HloInstruction* hlo) { in HandleCholesky()
724 Status HloCostAnalysis::HandleAllToAll(const HloInstruction* hlo) { in HandleAllToAll()
[all …]
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()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmemset_thunk.h36 const HloInstruction* hlo) in MemzeroThunk()
51 const HloInstruction* hlo) in Memset32BitValueThunk()
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.cc162 llvm::Value* HloToIrBindings::GetTypedIrValue(const HloInstruction& hlo, in GetTypedIrValue()
186 void HloToIrBindings::BindHloToIrValue(const HloInstruction& hlo, in BindHloToIrValue()
204 const HloInstruction& hlo, const HloInstruction& consumer, in BuffersInvariantWithinConsumer()
223 llvm_ir::IrArray HloToIrBindings::GetIrArray(const HloInstruction& hlo, in GetIrArray()
Dtree_reduction_rewriter.cc45 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce()
58 Status RewriteReduction(HloInstruction *hlo) { in RewriteReduction()
187 HloInstruction *hlo, const ReductionDimensions &reduction_dimensions, in RewriteBatchDimensionLargerThanTile()
Dcudnn_batchnorm_thunk.cc35 void CheckInputOutputPrimitivetypeAreValid(const HloInstruction* hlo) { in CheckInputOutputPrimitivetypeAreValid()
99 const BufferAllocation::Slice& output, const HloInstruction* hlo) in CudnnBatchNormForwardInferenceThunk()
148 const BufferAllocation::Slice& output_tuple, const HloInstruction* hlo) in CudnnBatchNormForwardTrainingThunk()
212 const BufferAllocation::Slice& output_tuple, const HloInstruction* hlo) in CudnnBatchNormBackwardThunk()
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.cc27 bool CanBeLoopFused(const HloInstruction& hlo) { in CanBeLoopFused()
45 bool IsNonComplexNonBatchedMatrixVectorDot(const HloInstruction* hlo) { in IsNonComplexNonBatchedMatrixVectorDot()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dllvm_irgen_test_base.h67 void MatchOptimizedHloWithShapes(absl::string_view hlo, in MatchOptimizedHloWithShapes()
Dllvm_irgen_test_base.cc86 void LlvmIrGenTestBase::MatchOptimizedHlo(absl::string_view hlo, in MatchOptimizedHlo()
100 absl::string_view hlo) { in GetOptimizedModule()

1234