| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | arithmetic_optimizer_test_utils.h | 33 void OptimizeAndPrune(GraphOptimizer* optimizer, GrapplerItem* item, in OptimizeAndPrune()
42 void DedupAndOptimizeTwiceAndPrune(GraphOptimizer* optimizer, in DedupAndOptimizeTwiceAndPrune()
57 void OptimizeTwice(GraphOptimizer* optimizer, GrapplerItem* item, in OptimizeTwice()
87 void DisableAddToAddNCombining(ArithmeticOptimizer* optimizer) { in DisableAddToAddNCombining()
91 void EnableOnlyAddToAddNCombining(ArithmeticOptimizer* optimizer) { in EnableOnlyAddToAddNCombining()
96 void EnableOnlyFoldConjugateIntoTranspose(ArithmeticOptimizer* optimizer) { in EnableOnlyFoldConjugateIntoTranspose()
101 void EnableOnlyFoldMultipleIntoConv(ArithmeticOptimizer* optimizer) { in EnableOnlyFoldMultipleIntoConv()
106 void EnableOnlyFoldTransposeIntoMatMul(ArithmeticOptimizer* optimizer) { in EnableOnlyFoldTransposeIntoMatMul()
111 void EnableOnlyHoistCommonFactor(ArithmeticOptimizer* optimizer) { in EnableOnlyHoistCommonFactor()
116 void EnableOnlyMinimizeBroadcasts(ArithmeticOptimizer* optimizer) { in EnableOnlyMinimizeBroadcasts()
[all …]
|
| H A D | arithmetic_optimizer_test.cc | 105 ArithmeticOptimizer optimizer; in TEST_F() local
131 ArithmeticOptimizer optimizer; in TEST_F() local
174 ArithmeticOptimizer optimizer; in TEST_F() local
208 ArithmeticOptimizer optimizer; in TEST_F() local
239 ArithmeticOptimizer optimizer; in TEST_F() local
269 ArithmeticOptimizer optimizer; in TEST_F() local
304 ArithmeticOptimizer optimizer; in TEST_F() local
364 ArithmeticOptimizer optimizer; in TEST_F() local
409 ArithmeticOptimizer optimizer; in TEST_F() local
478 ArithmeticOptimizer optimizer; in TEST_F() local
[all …]
|
| H A D | dependency_optimizer_test.cc | 59 DependencyOptimizer optimizer; in TEST_F() local
84 DependencyOptimizer optimizer; in TEST_F() local
119 DependencyOptimizer optimizer; in TEST_F() local
161 DependencyOptimizer optimizer; in TEST_F() local
219 DependencyOptimizer optimizer; in TEST_F() local
253 DependencyOptimizer optimizer; in TEST_F() local
273 DependencyOptimizer optimizer; in TEST_F() local
317 DependencyOptimizer optimizer; in TEST_F() local
349 DependencyOptimizer optimizer; in TEST_F() local
372 DependencyOptimizer optimizer; in TEST_F() local
[all …]
|
| H A D | function_optimizer_test.cc | 41 FunctionOptimizer optimizer(RewriterConfig::DEFAULT, true); in TEST_F() local
87 FunctionOptimizer optimizer(RewriterConfig::DEFAULT, true); in TEST_F() local
146 FunctionOptimizer optimizer(RewriterConfig::DEFAULT, true); in TEST_F() local
195 FunctionOptimizer optimizer(RewriterConfig::DEFAULT, true); in TEST_F() local
254 FunctionOptimizer optimizer(RewriterConfig::DEFAULT, true); in TEST_F() local
299 FunctionOptimizer optimizer(RewriterConfig::DEFAULT, true); in TEST_F() local
346 FunctionOptimizer optimizer(RewriterConfig::ON, true); in TEST_F() local
395 FunctionOptimizer optimizer(RewriterConfig::ON, true); in TEST_F() local
442 FunctionOptimizer optimizer(RewriterConfig::ON, true); in TEST_F() local
484 FunctionOptimizer optimizer(RewriterConfig::AGGRESSIVE, true); in TEST_F() local
[all …]
|
| H A D | constant_folding_test.cc | 84 ConstantFolding optimizer(/*cpu_device=*/nullptr); in SimpleNeutralElementTest() local
173 ConstantFolding optimizer(/*cpu_device=*/nullptr); in MulConvPushDownTest() local
246 ConstantFolding optimizer(/*cpu_device=*/nullptr); in PaddingWithZeroSize() local
292 ConstantFolding optimizer(/*cpu_device=*/nullptr); in TEST_F() local
337 ConstantFolding optimizer(/*cpu_device=*/nullptr); in TEST_F() local
416 ConstantFolding optimizer(/*cpu_device=*/nullptr); in TEST_F() local
509 ConstantFolding optimizer(/*cpu_device=*/nullptr); in TEST_F() local
575 ConstantFolding optimizer(/*cpu_device=*/nullptr); in TEST_F() local
827 ConstantFolding optimizer(/*cpu_device=*/nullptr); in TEST_F() local
1012 ConstantFolding optimizer(/*cpu_device=*/nullptr); in TEST_F() local
[all …]
|
| H A D | loop_optimizer_test.cc | 63 void EnableOnlyLoopInvariantNodeMotion(LoopOptimizer* optimizer) { in EnableOnlyLoopInvariantNodeMotion()
68 void EnableOnlyStackPushRemoval(LoopOptimizer* optimizer) { in EnableOnlyStackPushRemoval()
74 void DisableAllStages(LoopOptimizer* optimizer) { in DisableAllStages()
104 LoopOptimizer optimizer; in TEST_F() local
171 LoopOptimizer optimizer; in TEST_F() local
238 LoopOptimizer optimizer; in TEST_F() local
312 LoopOptimizer optimizer; in TEST_F() local
405 LoopOptimizer optimizer; in TEST_F() local
488 LoopOptimizer optimizer; in TEST_F() local
572 LoopOptimizer optimizer; in TEST_F() local
[all …]
|
| H A D | implementation_selector_test.cc | 50 std::unique_ptr<CustomGraphOptimizer> optimizer(new ImplementationSelector); in TEST_F() local
64 ImplementationSelector optimizer; in TEST_F() local
88 ImplementationSelector optimizer; in TEST_F() local
112 ImplementationSelector optimizer; in TEST_F() local
144 ImplementationSelector optimizer; in TEST_F() local
168 ImplementationSelector optimizer; in TEST_F() local
191 ImplementationSelector optimizer; in TEST_F() local
249 ImplementationSelector optimizer; in TEST_F() local
315 ImplementationSelector optimizer; in TEST_F() local
353 ImplementationSelector optimizer; in TEST_F() local
[all …]
|
| H A D | memory_optimizer_test.cc | 59 MemoryOptimizer optimizer(RewriterConfig::MANUAL); in TEST_F() local
99 MemoryOptimizer optimizer(RewriterConfig::MANUAL); in TEST_F() local
124 MemoryOptimizer optimizer(RewriterConfig::MANUAL, in TEST_F() local
164 MemoryOptimizer optimizer(RewriterConfig::RECOMPUTATION_HEURISTICS); in TEST_F() local
253 MemoryOptimizer optimizer(RewriterConfig::MANUAL); in TEST_F() local
323 MemoryOptimizer optimizer(RewriterConfig::SWAPPING_HEURISTICS); in TEST_F() local
374 MemoryOptimizer optimizer(RewriterConfig::SWAPPING_HEURISTICS); in TEST_F() local
412 MemoryOptimizer optimizer(RewriterConfig::SCHEDULING_HEURISTICS); in TEST_F() local
466 MemoryOptimizer optimizer(RewriterConfig::MANUAL); in TEST_F() local
498 MemoryOptimizer optimizer(RewriterConfig::MANUAL); in TEST_F() local
[all …]
|
| H A D | auto_mixed_precision_test.cc | 189 AutoMixedPrecision optimizer; in TestSimpleUnaryInferOp() local
228 AutoMixedPrecision optimizer; in TEST_F() local
266 AutoMixedPrecision optimizer; in TEST_F() local
309 AutoMixedPrecision optimizer; in TEST_F() local
361 AutoMixedPrecision optimizer; in TEST_F() local
406 AutoMixedPrecision optimizer; in TEST_F() local
447 AutoMixedPrecision optimizer; in TEST_F() local
492 AutoMixedPrecision optimizer; in TEST_F() local
535 AutoMixedPrecision optimizer; in TEST_F() local
592 AutoMixedPrecision optimizer; in TEST_F() local
[all …]
|
| H A D | meta_optimizer_test.cc | 146 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F() local
167 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F() local
189 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F() local
214 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F() local
232 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F() local
251 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F() local
271 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F() local
427 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F() local
528 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F() local
639 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F() local
[all …]
|
| H A D | generic_layout_optimizer_test.cc | 238 GenericLayoutOptimizer optimizer(REWRITER_CONFIG); in TEST_F() local
265 GenericLayoutOptimizer optimizer(REWRITER_CONFIG); in TEST_F() local
284 GenericLayoutOptimizer optimizer(REWRITER_CONFIG); in TEST_F() local
307 GenericLayoutOptimizer optimizer; in TEST_F() local
326 GenericLayoutOptimizer optimizer(REWRITER_CONFIG); in TEST_F() local
349 GenericLayoutOptimizer optimizer(REWRITER_CONFIG); in TEST_F() local
381 GenericLayoutOptimizer optimizer(REWRITER_CONFIG); in TEST_F() local
405 GenericLayoutOptimizer optimizer(REWRITER_CONFIG); in TEST_F() local
431 GenericLayoutOptimizer optimizer(REWRITER_CONFIG); in TEST_F() local
516 GenericLayoutOptimizer optimizer(REWRITER_CONFIG); in TEST_F() local
[all …]
|
| H A D | remapper_test.cc | 67 Remapper optimizer(RewriterConfig::ON); in TEST_F() local
101 Remapper optimizer(RewriterConfig::ON); in TEST_F() local
183 Remapper optimizer(RewriterConfig::AGGRESSIVE); // trust placeholders shape in TEST_F() local
288 Remapper optimizer(RewriterConfig::AGGRESSIVE); // trust placeholders shape in TEST_F() local
419 Remapper optimizer(RewriterConfig::AGGRESSIVE); // trust placeholders shape in TEST_F() local
559 Remapper optimizer(RewriterConfig::AGGRESSIVE); // trust placeholders shape in TEST_F() local
683 Remapper optimizer(RewriterConfig::ON); in RunTest() local
828 Remapper optimizer(RewriterConfig::ON); in RunTest() local
953 Remapper optimizer(RewriterConfig::ON); in RunTest() local
1050 Remapper optimizer(RewriterConfig::AGGRESSIVE); in TEST_F() local
[all …]
|
| H A D | debug_stripper_test.cc | 40 DebugStripper optimizer; in TEST_F() local
62 DebugStripper optimizer; in TEST_F() local
88 DebugStripper optimizer; in TEST_F() local
149 DebugStripper optimizer; in TEST_F() local
204 DebugStripper optimizer; in TEST_F() local
241 DebugStripper optimizer; in TEST_F() local
|
| /aosp_15_r20/external/deqp-deps/SPIRV-Tools/test/opt/ |
| D | c_interface_test.cpp | 40 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
76 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
124 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
173 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
222 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
283 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
347 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
411 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
476 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
509 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
[all …]
|
| /aosp_15_r20/external/angle/third_party/spirv-tools/src/test/opt/ |
| H A D | c_interface_test.cpp | 40 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
76 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
124 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
173 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
222 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
283 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
347 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
411 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
476 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
509 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
[all …]
|
| /aosp_15_r20/external/swiftshader/third_party/SPIRV-Tools/test/opt/ |
| H A D | c_interface_test.cpp | 40 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
76 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
124 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
173 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
222 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
283 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
347 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
411 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
476 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
509 auto optimizer = spvOptimizerCreate(SPV_ENV_UNIVERSAL_1_1); in TEST() local
[all …]
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | optim.cpp | 33 OptimizerClass optimizer(model->parameters(), options); in test_optimizer_xor() local
50 torch::Tensor labels) { in test_optimizer_xor()
122 auto optimizer = OptimizerClass(parameters.values(), options); in check_exact_values() local
164 auto optimizer = Adagrad(params, options); in TEST() local
229 MyOptimizer optimizer(parameters); in TEST() local
237 MyOptimizer optimizer(params); in TEST() local
257 MyOptimizer optimizer(linear->parameters()); in TEST() local
419 SGD optimizer(model->parameters(), 0.1); in TEST() local
454 SGD optimizer(parameters, 1.0); in TEST() local
474 LBFGS optimizer(std::vector<torch::Tensor>{}, 1.0); in TEST() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| H A D | make_deterministic_test.cc | 111 MakeDeterministic optimizer; in TEST_P() local
299 MakeDeterministic optimizer; in TEST_P() local
328 MakeDeterministic optimizer; in TEST_P() local
362 MakeDeterministic optimizer; in TEST_P() local
395 MakeDeterministic optimizer; in TEST_P() local
438 MakeDeterministic optimizer; in TEST_P() local
505 MakeDeterministic optimizer; in TEST_P() local
553 MakeDeterministic optimizer; in TEST_P() local
595 MakeDeterministic optimizer; in TEST_P() local
638 MakeDeterministic optimizer; in TEST() local
[all …]
|
| H A D | make_sloppy_test.cc | 52 MakeSloppy optimizer; in TEST() local
78 MakeSloppy optimizer; in TEST() local
102 MakeSloppy optimizer; in TEST() local
130 MakeSloppy optimizer; in TEST() local
156 MakeSloppy optimizer; in TEST() local
183 MakeSloppy optimizer; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/ |
| H A D | tpu_embedding_v1_correctness_test.py | 57 def _create_mid_level(self, optimizer=None): argument
65 def _get_slot_variable_creation_fn(self, optimizer): argument
179 gradients_wrt_video, optimizer, argument
197 gradients, optimizer, argument
207 gradients, optimizer, argument
222 gradients, optimizer, argument
243 gradients, optimizer, argument
|
| H A D | tpu_embedding_base_test.py | 160 def _create_mid_level(self, optimizer=None): argument
355 table_to_variable, optimizer, is_high_dimensional): argument
461 gradients_wrt_video, optimizer, argument
481 gradients, optimizer, argument
490 gradients, optimizer, argument
504 self, embedding_table_before, gradients, optimizer, variable): argument
534 gradients, optimizer, argument
555 gradients, optimizer, argument
|
| /aosp_15_r20/external/swiftshader/third_party/SPIRV-Tools/tools/opt/ |
| H A D | opt.cpp | 54 std::string GetListOfPassesAsString(const spvtools::Optimizer& optimizer) { in GetListOfPassesAsString()
65 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetLegalizationPasses() local
71 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetOptimizationPasses() local
77 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetSizePasses() local
632 spvtools::Optimizer* optimizer, const char** in_file, in ParseOconfigFlag()
714 spvtools::Optimizer* optimizer, const char** in_file, in ParseFlags()
848 spvtools::Optimizer optimizer(target_env); in main() local
|
| /aosp_15_r20/external/deqp-deps/SPIRV-Tools/tools/opt/ |
| D | opt.cpp | 54 std::string GetListOfPassesAsString(const spvtools::Optimizer& optimizer) { in GetListOfPassesAsString()
65 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetLegalizationPasses() local
71 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetOptimizationPasses() local
77 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetSizePasses() local
632 spvtools::Optimizer* optimizer, const char** in_file, in ParseOconfigFlag()
714 spvtools::Optimizer* optimizer, const char** in_file, in ParseFlags()
848 spvtools::Optimizer optimizer(target_env); in main() local
|
| /aosp_15_r20/external/angle/third_party/spirv-tools/src/tools/opt/ |
| H A D | opt.cpp | 54 std::string GetListOfPassesAsString(const spvtools::Optimizer& optimizer) { in GetListOfPassesAsString()
65 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetLegalizationPasses() local
71 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetOptimizationPasses() local
77 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetSizePasses() local
644 spvtools::Optimizer* optimizer, const char** in_file, in ParseOconfigFlag()
726 spvtools::Optimizer* optimizer, const char** in_file, in ParseFlags()
860 spvtools::Optimizer optimizer(target_env); in main() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_embedding_v2_utils_test.py | 30 def test_grad_clip_with_accumulation_off(self, optimizer): argument
40 def test_grad_clip_with_tuple(self, optimizer): argument
49 def test_grad_clip_with_single_value(self, optimizer): argument
58 def test_grad_clip_with_tuple_and_none(self, optimizer): argument
67 def test_equal_and_hash_function(self, optimizer): argument
|