| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 360 spmd_pipeline.AddPass<CallInliner>(); in OptimizeHloModule()
361 spmd_pipeline.AddPass<ZeroSizedHloElimination>(); in OptimizeHloModule()
362 spmd_pipeline.AddPass<ConditionalCanonicalizer>(); in OptimizeHloModule()
365 spmd_pipeline.AddPass<HloPassFix<HloPassPipeline>>("spmd-simplify"); in OptimizeHloModule()
367 spmd_simplify.AddPass<AlgebraicSimplifier>( in OptimizeHloModule()
370 spmd_simplify.AddPass<SortSimplifier>(); in OptimizeHloModule()
371 spmd_simplify.AddPass<TupleSimplifier>(); in OptimizeHloModule()
372 spmd_simplify.AddPass<ScatterSimplifier>(); in OptimizeHloModule()
373 spmd_simplify.AddPass<ScatterExpander>( in OptimizeHloModule()
375 spmd_simplify.AddPass<GatherSimplifier>(); in OptimizeHloModule()
[all …]
|
| D | nvptx_compiler.cc | 80 pipeline.AddPass<GpusolverRewriter>(); in OptimizeHloConvolutionCanonicalization()
81 pipeline.AddPass<GpuConvRewriter>(); in OptimizeHloConvolutionCanonicalization()
82 pipeline.AddPass<CudnnFusedConvRewriter>(); in OptimizeHloConvolutionCanonicalization()
83 pipeline.AddPass<GpuConvPaddingLegalization>(); in OptimizeHloConvolutionCanonicalization()
84 pipeline.AddPass<CudnnPadForConvolutions>( in OptimizeHloConvolutionCanonicalization()
86 pipeline.AddPass<CudnnVectorizeConvolutions>( in OptimizeHloConvolutionCanonicalization()
91 pipeline.AddPass<CallInliner>(); in OptimizeHloConvolutionCanonicalization()
92 pipeline.AddPass<TupleSimplifier>(); in OptimizeHloConvolutionCanonicalization()
96 pipeline.AddPass<HloPassFix<AlgebraicSimplifier>>(algsimp_options); in OptimizeHloConvolutionCanonicalization()
103 pipeline.AddPass<CudnnSimplifyPadding>(); in OptimizeHloConvolutionCanonicalization()
[all …]
|
| D | amdgpu_compiler.cc | 85 pipeline.AddPass<GpusolverRewriter>(); in OptimizeHloConvolutionCanonicalization()
86 pipeline.AddPass<GpuConvRewriter>(); in OptimizeHloConvolutionCanonicalization()
87 pipeline.AddPass<GpuConvPaddingLegalization>(); in OptimizeHloConvolutionCanonicalization()
92 pipeline.AddPass<CallInliner>(); in OptimizeHloConvolutionCanonicalization()
93 pipeline.AddPass<TupleSimplifier>(); in OptimizeHloConvolutionCanonicalization()
100 pipeline.AddPass<HloPassFix<AlgebraicSimplifier>>(options); in OptimizeHloConvolutionCanonicalization()
102 pipeline.AddPass<HloConstantFolding>(); in OptimizeHloConvolutionCanonicalization()
118 post_pipeline.AddPass<TriangularSolveRewriter>(); in OptimizeHloPostLayoutAssignment()
|
| D | horizontal_loop_fusion_test.cc | 190 fusion.AddPass<xla::gpu::GpuInstructionFusion>(/*may_duplicate=*/false); in TEST_F()
191 fusion.AddPass<xla::gpu::GpuInstructionFusion>(/*may_duplicate=*/true); in TEST_F()
495 iterative_h_fusion.AddPass<GpuHorizontalLoopFusion>(); in TEST_F()
496 iterative_h_fusion.AddPass<HloDCE>(); in TEST_F()
572 iterative_h_fusion.AddPass<GpuHorizontalLoopFusion>(); in TEST_F()
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/test/opt/ |
| D | pass_manager_test.cpp | 48 manager.AddPass<StripDebugInfoPass>(); in TEST()
52 manager.AddPass(MakeUnique<NullPass>()); in TEST()
57 manager.AddPass<StripDebugInfoPass>(); in TEST()
63 manager.AddPass<NullPassWithArgs>(1u); in TEST()
64 manager.AddPass<NullPassWithArgs>("null pass args"); in TEST()
65 manager.AddPass<NullPassWithArgs>(std::initializer_list<int>{1, 2}); in TEST()
66 manager.AddPass<NullPassWithArgs>(std::initializer_list<int>{1, 2}, 3); in TEST()
121 AddPass<AppendOpNopPass>(); in TEST_F()
122 AddPass<AppendOpNopPass>(); in TEST_F()
126 AddPass<AppendOpNopPass>(); in TEST_F()
[all …]
|
| D | pass_fixture.h | 222 void AddPass(Args&&... args) { in AddPass() function
223 manager_->AddPass<PassT>(std::forward<Args>(args)...); in AddPass()
|
| D | interface_var_sroa_test.cpp | 348 manager.AddPass<InterfaceVariableScalarReplacement>(); in RunPass()
|
| D | inline_test.cpp | 2927 AddPass<MergeReturnPass>(); in TEST_F()
2928 AddPass<InlineExhaustivePass>(); in TEST_F()
3027 AddPass<MergeReturnPass>(); in TEST_F()
3028 AddPass<InlineExhaustivePass>(); in TEST_F()
3190 AddPass<MergeReturnPass>(); in TEST_F()
3191 AddPass<InlineExhaustivePass>(); in TEST_F()
3318 AddPass<MergeReturnPass>(); in TEST_F()
3319 AddPass<InlineExhaustivePass>(); in TEST_F()
|
| /external/swiftshader/third_party/SPIRV-Tools/test/opt/ |
| D | pass_manager_test.cpp | 48 manager.AddPass<StripDebugInfoPass>(); in TEST()
52 manager.AddPass(MakeUnique<NullPass>()); in TEST()
57 manager.AddPass<StripDebugInfoPass>(); in TEST()
63 manager.AddPass<NullPassWithArgs>(1u); in TEST()
64 manager.AddPass<NullPassWithArgs>("null pass args"); in TEST()
65 manager.AddPass<NullPassWithArgs>(std::initializer_list<int>{1, 2}); in TEST()
66 manager.AddPass<NullPassWithArgs>(std::initializer_list<int>{1, 2}, 3); in TEST()
121 AddPass<AppendOpNopPass>(); in TEST_F()
122 AddPass<AppendOpNopPass>(); in TEST_F()
126 AddPass<AppendOpNopPass>(); in TEST_F()
[all …]
|
| D | pass_fixture.h | 222 void AddPass(Args&&... args) { in AddPass() function
223 manager_->AddPass<PassT>(std::forward<Args>(args)...); in AddPass()
|
| D | interface_var_sroa_test.cpp | 350 manager.AddPass<InterfaceVariableScalarReplacement>(); in RunPass()
|
| /external/deqp-deps/SPIRV-Tools/test/opt/ |
| D | pass_manager_test.cpp | 48 manager.AddPass<StripDebugInfoPass>(); in TEST()
52 manager.AddPass(MakeUnique<NullPass>()); in TEST()
57 manager.AddPass<StripDebugInfoPass>(); in TEST()
63 manager.AddPass<NullPassWithArgs>(1u); in TEST()
64 manager.AddPass<NullPassWithArgs>("null pass args"); in TEST()
65 manager.AddPass<NullPassWithArgs>(std::initializer_list<int>{1, 2}); in TEST()
66 manager.AddPass<NullPassWithArgs>(std::initializer_list<int>{1, 2}, 3); in TEST()
121 AddPass<AppendOpNopPass>(); in TEST_F()
122 AddPass<AppendOpNopPass>(); in TEST_F()
126 AddPass<AppendOpNopPass>(); in TEST_F()
[all …]
|
| D | pass_fixture.h | 222 void AddPass(Args&&... args) { in AddPass() function
223 manager_->AddPass<PassT>(std::forward<Args>(args)...); in AddPass()
|
| D | interface_var_sroa_test.cpp | 350 manager.AddPass<InterfaceVariableScalarReplacement>(); in RunPass()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 441 spmd_pipeline.AddPass<CallInliner>(); in RunHloPassesThroughLayoutAssn()
442 spmd_pipeline.AddPass<ZeroSizedHloElimination>(); in RunHloPassesThroughLayoutAssn()
443 spmd_pipeline.AddPass<ConditionalCanonicalizer>(); in RunHloPassesThroughLayoutAssn()
445 spmd_pipeline.AddPass<ShardingPropagation>( in RunHloPassesThroughLayoutAssn()
448 spmd_pipeline.AddPass<spmd::StatefulRngSpmdPartitioner>( in RunHloPassesThroughLayoutAssn()
452 spmd_pipeline.AddPass<ShardingRemover>(); in RunHloPassesThroughLayoutAssn()
453 spmd_pipeline.AddPass<HloDCE>(); in RunHloPassesThroughLayoutAssn()
461 pipeline.AddPass<OperandUpcaster>(); in RunHloPassesThroughLayoutAssn()
462 pipeline.AddPass<ResultCaster>(); in RunHloPassesThroughLayoutAssn()
465 pipeline.AddPass<RngExpander>(); in RunHloPassesThroughLayoutAssn()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | stateful_rng_spmd_partitioner_test.cc | 48 pass.AddPass<HloVerifier>(/*layout_sensitive=*/false, in PartitionComputation()
53 pass.AddPass<ShardingPropagation>(/*is_spmd=*/true); in PartitionComputation()
54 pass.AddPass<StatefulRngSpmdPartitioner>(num_partitions, in PartitionComputation()
56 pass.AddPass<HloVerifier>(/*layout_sensitive=*/false, in PartitionComputation()
85 pipeline.AddPass<RngExpander>(); in TEST_F()
109 pipeline.AddPass<RngExpander>(); in TEST_F()
|
| D | canonicalize_all_gather_for_cse_test.cc | 41 pipeline.AddPass<CanonicalizeAllGatherForCSE>(); in RunPass()
47 pipeline.AddPass<CanonicalizeAllGatherForCSE>(); in RunPassOnModule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_spmd_partitioner_test.cc | 48 pass.AddPass<HloVerifier>(/*layout_sensitive=*/false, in PartitionComputation()
53 pass.AddPass<ShardingPropagation>(/*is_spmd=*/true); in PartitionComputation()
54 pass.AddPass<GpuSpmdPartitioner>(num_partitions, /*num_replicas=*/1); in PartitionComputation()
55 pass.AddPass<HloVerifier>(/*layout_sensitive=*/false, in PartitionComputation()
84 pipeline.AddPass<RngExpander>(); in TEST_F()
108 pipeline.AddPass<RngExpander>(); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 85 pipeline.AddPass<DynamicIndexSplitter>(); in RunHloOptimization()
86 pipeline.AddPass<CholeskyExpander>(); in RunHloOptimization()
87 pipeline.AddPass<QrExpander>(); in RunHloOptimization()
88 pipeline.AddPass<EighExpander>(); in RunHloOptimization()
89 pipeline.AddPass<ComparisonExpander>(); in RunHloOptimization()
90 pipeline.AddPass<TriangularSolveExpander>(); in RunHloOptimization()
91 pipeline.AddPass<BatchNormExpander>( in RunHloOptimization()
95 pipeline.AddPass<LayoutAssignment>( in RunHloOptimization()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | despecializer.cc | 26 pipeline_.AddPass<HloDescheduler>(); in Despecializer()
27 pipeline_.AddPass<ControlDepRemover>(); in Despecializer()
28 pipeline_.AddPass<Defuser>(); in Despecializer()
29 pipeline_.AddPass<BFloat16MixedPrecisionRemoval>(); in Despecializer()
|
| D | hlo_pass_pipeline_test.cc | 150 pipeline.AddPass<FooToBarModulePass>(); in TEST_F()
173 pipeline.AddPass<FooToBarModulePass>(); in TEST_F()
200 pipeline.AddPass<ReverseStringModulePass>(); in TEST_F()
236 pipeline.AddPass<ReverseStringModulePass>(); in TEST_F()
274 pipeline.AddPass<BazToQuxModuleGroupPass>(); in TEST_F()
275 pipeline.AddPass<FooToBarModulePass>(); in TEST_F()
319 pipeline.AddPass<FooToBarModulePass>(); in TEST_F()
357 pipeline.AddPass<BazToQuxModuleGroupPass>(); in TEST_F()
373 pipeline.AddPass<BazToQuxModuleGroupPass>(); in TEST_F()
374 pipeline.AddPass<FooToBarModulePass>(); in TEST_F()
|
| D | while_loop_concat_code_motion.cc | 1040 pipeline.AddPass<TupleSimplifier>(); in Run()
1041 pipeline.AddPass<HloDCE>(); in Run()
1042 pipeline.AddPass<WhileLoopSimplifier>(); in Run()
1043 pipeline.AddPass<TupleSimplifier>(); in Run()
1044 pipeline.AddPass<HloDCE>(); in Run()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | pass_manager.h | 55 void AddPass(std::unique_ptr<Pass> pass);
60 void AddPass(Args&&... args);
132 inline void PassManager::AddPass(std::unique_ptr<Pass> pass) { in AddPass() function
137 inline void PassManager::AddPass(Args&&... args) { in AddPass() function
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| D | pass_manager.h | 55 void AddPass(std::unique_ptr<Pass> pass);
60 void AddPass(Args&&... args);
132 inline void PassManager::AddPass(std::unique_ptr<Pass> pass) { in AddPass() function
137 inline void PassManager::AddPass(Args&&... args) { in AddPass() function
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/ |
| D | pass_manager.h | 55 void AddPass(std::unique_ptr<Pass> pass);
60 void AddPass(Args&&... args);
132 inline void PassManager::AddPass(std::unique_ptr<Pass> pass) { in AddPass() function
137 inline void PassManager::AddPass(Args&&... args) { in AddPass() function
|