Searched refs:DynamicDimensionInferenceVisitor (Results 1 – 2 of 2) sorted by relevance
26 class DynamicDimensionInferenceVisitor : public DfsHloVisitorWithDefault { class28 explicit DynamicDimensionInferenceVisitor( in DynamicDimensionInferenceVisitor() function in xla::DynamicDimensionInferenceVisitor38 DynamicDimensionInferenceVisitor visitor(param_bindings, parent); in Run()98 Status DynamicDimensionInferenceVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction()109 Status DynamicDimensionInferenceVisitor::HandleGetTupleElement( 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()[all …]
49 friend class DynamicDimensionInferenceVisitor; variable