Searched refs:GetOrCreateSubComputationFromRegion (Results 1 – 2 of 2) sorted by relevance
1682 GetOrCreateSubComputationFromRegion(&fusion.getRegion(), in EmitLoopFusion()1795 GetOrCreateSubComputationFromRegion(&fusion_op.getRegion(), in EmitFusion()2079 GetOrCreateSubComputationFromRegion(&select_and_scatter_op.getSelect(), in EmitSelectAndScatter()2136 GetOrCreateSubComputationFromRegion(&select_and_scatter_op.getScatter(), in EmitSelectAndScatter()2289 GetOrCreateSubComputationFromRegion(&scatter.getUpdateComputation(), in EmitScatter()2444 IrEmitterUnnested::GetOrCreateSubComputationFromRegion(mlir::Region* region, in GetOrCreateSubComputationFromRegion() function in xla::gpu::IrEmitterUnnested2705 GetOrCreateSubComputationFromRegion( in EmitSort()3274 *GetOrCreateSubComputationFromRegion(&fusion.getRegion(), in BuildFusedInitializerThunk()3521 *GetOrCreateSubComputationFromRegion(&fusion.getRegion(), in EmitTileElementForTranspose()4911 GetOrCreateSubComputationFromRegion(&fusion.getRegion(), in EmitUnnestedReduction()[all …]
755 StatusOr<HloComputation*> GetOrCreateSubComputationFromRegion(