Index

A

AlgebraicSimplifier (class in hloenv.HloPass) AllGatherBroadcastReorder (class in hloenv.HloPass) AllGatherCombiner (class in hloenv.HloPass) AllReduceBlueConnect (class in hloenv.HloPass) AllReduceCombiner (class in hloenv.HloPass)

AllReduceContiguous (class in hloenv.HloPass) AllReduceFolder (class in hloenv.HloPass) AllReduceReassociate (class in hloenv.HloPass) AllToAllDecomposer (class in hloenv.HloPass) apply_alternatives() (hloenv.HloEnv method) AsyncCollectiveCreator (class in hloenv.HloPass)

B

BatchNormExpander (class in hloenv.HloPass) BF16 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute)

BFloat16Normalization (class in hloenv.HloPass) BitcastDtypesExpander (class in hloenv.HloPass)

C

C128 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) C64 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) CallInliner (class in hloenv.HloPass) clone() (hloenv.HloModule method) clone_hlo() (hloenv.HloEnv method) CollectivesScheduleLinearizer (class in hloenv.HloPass) ComparisonExpander (class in hloenv.HloPass) computation_count (hloenv.HloModule attribute)

compute_durations (hloenv.EvaluationResult attribute) ConditionalCanonicalizer (class in hloenv.HloPass) ConditionalSimplifier (class in hloenv.HloPass) config (hloenv.HloModule attribute) Convolution4DExpander (class in hloenv.HloPass) cost_analysis() (hloenv.HloModule method) CublasPadForGemms (class in hloenv.HloPass) CublasPadForGemms.PrimitiveType (class in hloenv.HloPass) CudnnFusedConvRewriter (class in hloenv.HloPass) CudnnPadForConvolutions (class in hloenv.HloPass) CudnnVectorizeConvolutions (class in hloenv.HloPass)

D

DotDecomposer (class in hloenv.HloPass) DotMerger (class in hloenv.HloPass) durations (hloenv.EvaluationResult attribute)

DynamicDimensionSimplifier (class in hloenv.HloPass) DynamicIndexSplitter (class in hloenv.HloPass) DynamicPadder (class in hloenv.HloPass) DynamicPadder.ShapeCheckMode (class in hloenv.HloPass)

E

EighExpander (class in hloenv.HloPass) evaluate() (hloenv.HloEnv method) EvaluationResult (class in hloenv)

export_hlo_to_str() (hloenv.HloEnv method) extract_fusions_as_module() (hloenv.HloModule method) extract_instructions_as_module() (hloenv.HloModule method) extract_random_submodule() (hloenv.HloModule method)

F

F16 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) F32 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) F64 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute)

FlattenCallGraph (class in hloenv.HloPass) full_durations (hloenv.EvaluationResult attribute) FusionBitcastLift (class in hloenv.HloPass) FusionMerger (class in hloenv.HloPass)

G

GatherExpander (class in hloenv.HloPass) GatherExpander.Mode (class in hloenv.HloPass) GemmAlgorithmPicker (class in hloenv.HloPass) GemmBroadcastFoldingRewriter (class in hloenv.HloPass) GemmRewriter (class in hloenv.HloPass) GeneralFusion (class in hloenv.HloPass) get_graph_load_errors() (hloenv.HloGraph method) get_hlo_graph() (hloenv.HloEnv method) get_hlo_module() (hloenv.HloEnv method) get_hlo_module_hash() (hloenv.HloEnv method)

GpuConvAlgorithmPicker (class in hloenv.HloPass) GpuConvPaddingLegalization (class in hloenv.HloPass) GpuConvRewriter (class in hloenv.HloPass) GpuHorizontalInputFusion (class in hloenv.HloPass) GpuHorizontalLoopFusion (class in hloenv.HloPass) GpuInstructionFusion (class in hloenv.HloPass) GpuLayoutAssignment (class in hloenv.HloPass) GpuMultiOutputFusion (class in hloenv.HloPass) GpuScatterExpander (class in hloenv.HloPass) GpusolverRewriter (class in hloenv.HloPass) GpuTreeReductionRewriter (class in hloenv.HloPass)

H

has_equal_output() (hloenv.HloEnv method) has_equal_output_as() (hloenv.HloEnv method) hash() (hloenv.HloGraph method) (hloenv.HloModule method) HloConstantFolding (class in hloenv.HloPass) HloCSE (class in hloenv.HloPass) HloDCE (class in hloenv.HloPass)

HloEnv (class in hloenv) hloenv (module) hloenv.HloPass (module) HloGraph (class in hloenv) HloModule (class in hloenv) HloPassInterface (class in hloenv.HloPass) HloVerifier (class in hloenv.HloPass)

I

instruction_count (hloenv.HloModule attribute)

is_bef_enabled() (hloenv.HloModule method)

K

kCompileTime (hloenv.HloPass.DynamicPadder attribute) (hloenv.HloPass.DynamicPadder.ShapeCheckMode attribute) kEliminateAllGathers (hloenv.HloPass.GatherExpander attribute) (hloenv.HloPass.GatherExpander.Mode attribute) kEliminateAllScatters (hloenv.HloPass.ScatterExpander attribute) (hloenv.HloPass.ScatterExpander.Mode attribute) kEliminateSimpleGathers (hloenv.HloPass.GatherExpander attribute) (hloenv.HloPass.GatherExpander.Mode attribute) kEliminateSimpleScatters (hloenv.HloPass.ScatterExpander attribute) (hloenv.HloPass.ScatterExpander.Mode attribute)

kExp (hloenv.HloPass.LogisticExpander attribute) (hloenv.HloPass.LogisticExpander.LogisticExpansionType attribute) kIgnore (hloenv.HloPass.DynamicPadder attribute) (hloenv.HloPass.DynamicPadder.ShapeCheckMode attribute) kInvalid (hloenv.HloPass.DynamicPadder attribute) (hloenv.HloPass.DynamicPadder.ShapeCheckMode attribute) kRuntimeTime (hloenv.HloPass.DynamicPadder attribute) (hloenv.HloPass.DynamicPadder.ShapeCheckMode attribute) kTanh (hloenv.HloPass.LogisticExpander attribute) (hloenv.HloPass.LogisticExpander.LogisticExpansionType attribute)

L

load_hlo() (hloenv.HloEnv method)

LogisticExpander (class in hloenv.HloPass) LogisticExpander.LogisticExpansionType (class in hloenv.HloPass)

N

name (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) (hloenv.HloPass.DynamicPadder.ShapeCheckMode attribute) (hloenv.HloPass.GatherExpander.Mode attribute) (hloenv.HloPass.LogisticExpander.LogisticExpansionType attribute) (hloenv.HloPass.RngBitGeneratorExpander.RandomAlgorithm attribute) (hloenv.HloPass.ScatterExpander.Mode attribute)

O

OPAQUE_TYPE (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) OperandUpcaster (class in hloenv.HloPass)

OptimizationBarrierExpander (class in hloenv.HloPass) optimize_hlo_module() (hloenv.HloEnv method) output (hloenv.EvaluationResult attribute)

P

PRED (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute)

prepare_for_eval() (hloenv.HloEnv method) PRIMITIVE_TYPE_INVALID (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute)

Q

QrExpander (class in hloenv.HloPass)

R

RealImagExpander (class in hloenv.HloPass) ReduceScatterCombiner (class in hloenv.HloPass) ReduceScatterCreator (class in hloenv.HloPass) ReductionDegenerateDimRemover (class in hloenv.HloPass) ReductionDimensionGrouper (class in hloenv.HloPass) ReductionLayoutNormalizer (class in hloenv.HloPass) ReductionSplitter (class in hloenv.HloPass) ReshapeMover (class in hloenv.HloPass) ResultCaster (class in hloenv.HloPass)

RNG_DEFAULT (hloenv.HloPass.RngBitGeneratorExpander attribute) (hloenv.HloPass.RngBitGeneratorExpander.RandomAlgorithm attribute) RNG_PHILOX (hloenv.HloPass.RngBitGeneratorExpander attribute) (hloenv.HloPass.RngBitGeneratorExpander.RandomAlgorithm attribute) RNG_THREE_FRY (hloenv.HloPass.RngBitGeneratorExpander attribute) (hloenv.HloPass.RngBitGeneratorExpander.RandomAlgorithm attribute) RngBitGeneratorExpander (class in hloenv.HloPass) RngBitGeneratorExpander.RandomAlgorithm (class in hloenv.HloPass) RngExpander (class in hloenv.HloPass) run() (hloenv.HloEnv method)

S

S16 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) S32 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) S64 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) S8 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute)

ScatterExpander (class in hloenv.HloPass) ScatterExpander.Mode (class in hloenv.HloPass) ShardingPropagation (class in hloenv.HloPass) ShardingRemover (class in hloenv.HloPass) SliceSinker (class in hloenv.HloPass) SortSimplifier (class in hloenv.HloPass) StableSortExpander (class in hloenv.HloPass) StatefulRngSpmdPartitioner (class in hloenv.HloPass)

T

to_string() (hloenv.HloModule method) TOKEN (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) TransposeFolding (class in hloenv.HloPass)

TriangularSolveRewriter (class in hloenv.HloPass) TUPLE (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) TupleSimplifier (class in hloenv.HloPass)

U

U16 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) U32 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute)

U64 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) U8 (hloenv.HloPass.CublasPadForGemms attribute) (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute)

V

value (hloenv.HloPass.CublasPadForGemms.PrimitiveType attribute) (hloenv.HloPass.DynamicPadder.ShapeCheckMode attribute) (hloenv.HloPass.GatherExpander.Mode attribute) (hloenv.HloPass.LogisticExpander.LogisticExpansionType attribute) (hloenv.HloPass.RngBitGeneratorExpander.RandomAlgorithm attribute) (hloenv.HloPass.ScatterExpander.Mode attribute)

VariadicOpSplitter (class in hloenv.HloPass)

W

WhileLoopConstantSinking (class in hloenv.HloPass)

WhileLoopSimplifier (class in hloenv.HloPass) WhileLoopTripCountAnnotator (class in hloenv.HloPass)

Z

ZeroSizedHloElimination (class in hloenv.HloPass)