Searched refs:GetOrCreateSubComputationFromRegion (Results 1 – 2 of 2) sorted by relevance
1791 GetOrCreateSubComputationFromRegion(&fusion.region(), in EmitLoopFusionFromMlir()1882 GetOrCreateSubComputationFromRegion(&fusion_op.region(), in HandleFusion()2052 GetOrCreateSubComputationFromRegion(&fusion_op.region(), in HandleFusion()2365 GetOrCreateSubComputationFromRegion(&select_and_scatter_op.select(), in EmitSelectAndScatterFromMlir()2410 GetOrCreateSubComputationFromRegion(&select_and_scatter_op.scatter(), in EmitSelectAndScatterFromMlir()2576 GetOrCreateSubComputationFromRegion(&scatter.update_computation(), in EmitScatter()2735 IrEmitterUnnested::GetOrCreateSubComputationFromRegion(mlir::Region* region, in GetOrCreateSubComputationFromRegion() function in xla::gpu::IrEmitterUnnested3005 GetOrCreateSubComputationFromRegion( in EmitSortFromMlir()3723 *GetOrCreateSubComputationFromRegion(&fusion.region(), in BuildFusedInitializerThunkForMlir()4151 *GetOrCreateSubComputationFromRegion(&fusion.region(), in EmitTileElementForFusion()[all …]
728 StatusOr<HloComputation*> GetOrCreateSubComputationFromRegion(