| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test_utils.h | 33 void OptimizeAndPrune(GraphOptimizer* optimizer, GrapplerItem* item, in OptimizeAndPrune() argument
35 TF_EXPECT_OK(optimizer->Optimize(nullptr, *item, output)); in OptimizeAndPrune()
42 void DedupAndOptimizeTwiceAndPrune(GraphOptimizer* optimizer, in DedupAndOptimizeTwiceAndPrune() argument
47 TF_EXPECT_OK(optimizer->Optimize(nullptr, *item, output)); in DedupAndOptimizeTwiceAndPrune()
50 TF_EXPECT_OK(optimizer->Optimize(nullptr, *item, output)); in DedupAndOptimizeTwiceAndPrune()
57 void OptimizeTwice(GraphOptimizer* optimizer, GrapplerItem* item, in OptimizeTwice() argument
59 TF_EXPECT_OK(optimizer->Optimize(nullptr, *item, output)); in OptimizeTwice()
62 TF_EXPECT_OK(optimizer->Optimize(nullptr, *item, output)); in OptimizeTwice()
67 void OptimizeTwiceAndPrune(GraphOptimizer* optimizer, GrapplerItem* item,
69 TF_EXPECT_OK(optimizer->Optimize(nullptr, *item, output));
[all …]
|
| D | layout_optimizer_test.cc | 236 LayoutOptimizer optimizer; in TEST_F() local
239 Status status = optimizer.Optimize(virtual_cluster_.get(), item, &output); in TEST_F()
271 LayoutOptimizer optimizer; in TEST_F() local
274 Status status = optimizer.Optimize(virtual_cluster_.get(), item, &output); in TEST_F()
293 LayoutOptimizer optimizer; in TEST_F() local
295 Status status = optimizer.Optimize(virtual_cluster_.get(), item, &output); in TEST_F()
306 LayoutOptimizer optimizer; in TEST_F() local
308 Status status = optimizer.Optimize(virtual_cluster_.get(), item, &output); in TEST_F()
319 LayoutOptimizer optimizer; in TEST_F() local
321 Status status = optimizer.Optimize(virtual_cluster_.get(), item, &output); in TEST_F()
[all …]
|
| D | implementation_selector_test.cc | 50 std::unique_ptr<CustomGraphOptimizer> optimizer(new ImplementationSelector); in TEST_F() local
51 ASSERT_NE(nullptr, optimizer); in TEST_F()
52 TF_ASSERT_OK(optimizer->Init()); in TEST_F()
55 const Status status = optimizer->Optimize(nullptr, item, &output); in TEST_F()
64 ImplementationSelector optimizer; in TEST_F() local
75 TF_EXPECT_OK(optimizer.Optimize(nullptr, item, &output)); in TEST_F()
88 ImplementationSelector optimizer; in TEST_F() local
99 TF_EXPECT_OK(optimizer.Optimize(nullptr, item, &output)); in TEST_F()
112 ImplementationSelector optimizer; in TEST_F() local
127 TF_EXPECT_OK(optimizer.Optimize(nullptr, item, &output)); in TEST_F()
[all …]
|
| D | arithmetic_optimizer_test.cc | 102 ArithmeticOptimizer optimizer; in TEST_F() local
104 Status status = optimizer.Optimize(nullptr, item, &output); in TEST_F()
125 ArithmeticOptimizer optimizer; in TEST_F() local
126 EnableOnlyReplaceMulWithSquare(&optimizer); in TEST_F()
127 OptimizeAndPrune(&optimizer, &item, &output); in TEST_F()
170 ArithmeticOptimizer optimizer; in TEST_F() local
171 EnableOnlyRemoveInvolution(&optimizer); in TEST_F()
172 OptimizeAndPrune(&optimizer, &item, &output); in TEST_F()
204 ArithmeticOptimizer optimizer; in TEST_F() local
205 EnableOnlyRemoveInvolution(&optimizer); in TEST_F()
[all …]
|
| /external/angle/third_party/vulkan-deps/glslang/src/SPIRV/ |
| D | SpvTools.cpp | 178 spvtools::Optimizer optimizer(target_env); in SpirvToolsTransform() local
179 optimizer.SetMessageConsumer(OptimizerMesssageConsumer); in SpirvToolsTransform()
186 optimizer.RegisterPass(spvtools::CreateStripDebugInfoPass()); in SpirvToolsTransform()
188 optimizer.RegisterPass(spvtools::CreateWrapOpKillPass()); in SpirvToolsTransform()
189 optimizer.RegisterPass(spvtools::CreateDeadBranchElimPass()); in SpirvToolsTransform()
190 optimizer.RegisterPass(spvtools::CreateMergeReturnPass()); in SpirvToolsTransform()
191 optimizer.RegisterPass(spvtools::CreateInlineExhaustivePass()); in SpirvToolsTransform()
192 optimizer.RegisterPass(spvtools::CreateEliminateDeadFunctionsPass()); in SpirvToolsTransform()
193 optimizer.RegisterPass(spvtools::CreateScalarReplacementPass()); in SpirvToolsTransform()
194 optimizer.RegisterPass(spvtools::CreateLocalAccessChainConvertPass()); in SpirvToolsTransform()
[all …]
|
| /external/deqp-deps/glslang/SPIRV/ |
| D | SpvTools.cpp | 178 spvtools::Optimizer optimizer(target_env); in SpirvToolsTransform() local
179 optimizer.SetMessageConsumer(OptimizerMesssageConsumer); in SpirvToolsTransform()
186 optimizer.RegisterPass(spvtools::CreateStripDebugInfoPass()); in SpirvToolsTransform()
188 optimizer.RegisterPass(spvtools::CreateWrapOpKillPass()); in SpirvToolsTransform()
189 optimizer.RegisterPass(spvtools::CreateDeadBranchElimPass()); in SpirvToolsTransform()
190 optimizer.RegisterPass(spvtools::CreateMergeReturnPass()); in SpirvToolsTransform()
191 optimizer.RegisterPass(spvtools::CreateInlineExhaustivePass()); in SpirvToolsTransform()
192 optimizer.RegisterPass(spvtools::CreateEliminateDeadFunctionsPass()); in SpirvToolsTransform()
193 optimizer.RegisterPass(spvtools::CreateScalarReplacementPass()); in SpirvToolsTransform()
194 optimizer.RegisterPass(spvtools::CreateLocalAccessChainConvertPass()); in SpirvToolsTransform()
[all …]
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | tracking_util_with_v1_optimizers_test.py | 82 optimizer = adam.AdamOptimizer(0.001)
85 optimizer=optimizer, model=model, optimizer_step=optimizer_step)
87 optimizer.minimize(
90 optimizer.minimize(
94 train_op = optimizer.minimize(
96 optimizer.minimize(
168 optimizer.get_slot(
187 optimizer = adam.AdamOptimizer(0.001)
189 optimizer=optimizer, model=model)
192 optimizer.minimize(
[all …]
|
| D | tracking_util_test.py | 96 model.compile(optimizer="sgd", loss="mse")
111 optimizer = adam.Adam(0.001)
114 optimizer=optimizer, model=model, step=step)
121 optimizer.apply_gradients(zip(gradients, variables)),
128 optimizer.apply_gradients(zip(gradients, variables))
202 optimizer = adam.Adam(0.001)
204 optimizer=optimizer, model=model)
210 train_op = optimizer.apply_gradients(zip(gradients, variables))
217 m_bias_slot = optimizer.get_slot(model._named_dense.variables[1], "m")
223 sorted(optimizer.variables(), key=lambda v: v.name))
[all …]
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util_with_v1_optimizers_test.py | 48 optimizer = adam.AdamOptimizer(0.1)
50 optimizer.minimize(root.var.read_value)
52 train_op = optimizer.minimize(root.var)
57 trackable_utils.Checkpoint(root=root, optimizer=optimizer)))
61 root.optimizer = optimizer
63 self.evaluate(state_ops.assign(optimizer.get_slot(name="m", var=root.var),
78 new_root.optimizer = adam.AdamOptimizer(0.1)
87 new_root.optimizer.get_slot(name="m", var=new_root.var)))
89 self.assertIs(new_root.optimizer.get_slot(name="m", var=new_root.var),
92 new_root.optimizer.minimize(new_root.var.read_value)
[all …]
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | optimizers_test.py | 50 def _test_optimizer(self, optimizer, target=0.75): argument
61 optimizer=optimizer,
65 keras.backend.get_value(model.optimizer.iterations), 0)
68 keras.backend.get_value(model.optimizer.iterations),
71 config = keras.optimizers.serialize(optimizer)
98 optimizer=optimizer,
102 keras.backend.get_value(model.optimizer.iterations),
106 keras.backend.get_value(model.optimizer.iterations), 127)
169 optimizer = optimizer_v1.TFOptimizer(AdamOptimizer(0.01))
176 optimizer=optimizer,
[all …]
|
| /external/apache-commons-math/src/main/java/org/apache/commons/math/optimization/ |
| D | MultiStartUnivariateRealOptimizer.java | 45 private final UnivariateRealOptimizer optimizer; field in MultiStartUnivariateRealOptimizer
79 public MultiStartUnivariateRealOptimizer(final UnivariateRealOptimizer optimizer, in MultiStartUnivariateRealOptimizer() argument
82 this.optimizer = optimizer; in MultiStartUnivariateRealOptimizer()
103 return optimizer.getAbsoluteAccuracy(); in getAbsoluteAccuracy()
128 return optimizer.getRelativeAccuracy(); in getRelativeAccuracy()
133 optimizer.resetAbsoluteAccuracy(); in resetAbsoluteAccuracy()
138 optimizer.resetMaximalIterationCount(); in resetMaximalIterationCount()
143 optimizer.resetRelativeAccuracy(); in resetRelativeAccuracy()
148 optimizer.setAbsoluteAccuracy(accuracy); in setAbsoluteAccuracy()
163 optimizer.setRelativeAccuracy(accuracy); in setRelativeAccuracy()
[all …]
|
| D | MultiStartDifferentiableMultivariateRealOptimizer.java | 44 private final DifferentiableMultivariateRealOptimizer optimizer; field in MultiStartDifferentiableMultivariateRealOptimizer
78 …artDifferentiableMultivariateRealOptimizer(final DifferentiableMultivariateRealOptimizer optimizer, in MultiStartDifferentiableMultivariateRealOptimizer() argument
81 this.optimizer = optimizer; in MultiStartDifferentiableMultivariateRealOptimizer()
163 optimizer.setConvergenceChecker(checker); in setConvergenceChecker()
168 return optimizer.getConvergenceChecker(); in getConvergenceChecker()
186 optimizer.setMaxIterations(maxIterations - totalIterations); in optimize()
187 optimizer.setMaxEvaluations(maxEvaluations - totalEvaluations); in optimize()
188 optima[i] = optimizer.optimize(f, goalType, in optimize()
196 totalIterations += optimizer.getIterations(); in optimize()
197 totalEvaluations += optimizer.getEvaluations(); in optimize()
[all …]
|
| D | MultiStartDifferentiableMultivariateVectorialOptimizer.java | 47 private final DifferentiableMultivariateVectorialOptimizer optimizer; field in MultiStartDifferentiableMultivariateVectorialOptimizer
82 final DifferentiableMultivariateVectorialOptimizer optimizer, in MultiStartDifferentiableMultivariateVectorialOptimizer() argument
85 this.optimizer = optimizer; in MultiStartDifferentiableMultivariateVectorialOptimizer()
167 optimizer.setConvergenceChecker(checker); in setConvergenceChecker()
172 return optimizer.getConvergenceChecker(); in getConvergenceChecker()
190 optimizer.setMaxIterations(maxIterations - totalIterations); in optimize()
191 optimizer.setMaxEvaluations(maxEvaluations - totalEvaluations); in optimize()
192 optima[i] = optimizer.optimize(f, target, weights, in optimize()
200 totalIterations += optimizer.getIterations(); in optimize()
201 totalEvaluations += optimizer.getEvaluations(); in optimize()
[all …]
|
| D | MultiStartMultivariateRealOptimizer.java | 44 private final MultivariateRealOptimizer optimizer; field in MultiStartMultivariateRealOptimizer
75 public MultiStartMultivariateRealOptimizer(final MultivariateRealOptimizer optimizer, in MultiStartMultivariateRealOptimizer() argument
78 this.optimizer = optimizer; in MultiStartMultivariateRealOptimizer()
153 optimizer.setConvergenceChecker(checker); in setConvergenceChecker()
158 return optimizer.getConvergenceChecker(); in getConvergenceChecker()
175 optimizer.setMaxIterations(maxIterations - totalIterations); in optimize()
176 optimizer.setMaxEvaluations(maxEvaluations - totalEvaluations); in optimize()
177 optima[i] = optimizer.optimize(f, goalType, in optimize()
185 totalIterations += optimizer.getIterations(); in optimize()
186 totalEvaluations += optimizer.getEvaluations(); in optimize()
|
| /external/swiftshader/third_party/SPIRV-Tools/tools/opt/ |
| D | opt.cpp | 54 std::string GetListOfPassesAsString(const spvtools::Optimizer& optimizer) { in GetListOfPassesAsString() argument
56 for (const auto& name : optimizer.GetPassNames()) { in GetListOfPassesAsString()
65 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetLegalizationPasses() local
66 optimizer.RegisterLegalizationPasses(); in GetLegalizationPasses()
67 return GetListOfPassesAsString(optimizer); in GetLegalizationPasses()
71 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetOptimizationPasses() local
72 optimizer.RegisterPerformancePasses(); in GetOptimizationPasses()
73 return GetListOfPassesAsString(optimizer); in GetOptimizationPasses()
77 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetSizePasses() local
78 optimizer.RegisterSizePasses(); in GetSizePasses()
[all …]
|
| /external/deqp-deps/SPIRV-Tools/tools/opt/ |
| D | opt.cpp | 54 std::string GetListOfPassesAsString(const spvtools::Optimizer& optimizer) { in GetListOfPassesAsString() argument
56 for (const auto& name : optimizer.GetPassNames()) { in GetListOfPassesAsString()
65 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetLegalizationPasses() local
66 optimizer.RegisterLegalizationPasses(); in GetLegalizationPasses()
67 return GetListOfPassesAsString(optimizer); in GetLegalizationPasses()
71 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetOptimizationPasses() local
72 optimizer.RegisterPerformancePasses(); in GetOptimizationPasses()
73 return GetListOfPassesAsString(optimizer); in GetOptimizationPasses()
77 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetSizePasses() local
78 optimizer.RegisterSizePasses(); in GetSizePasses()
[all …]
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/tools/opt/ |
| D | opt.cpp | 54 std::string GetListOfPassesAsString(const spvtools::Optimizer& optimizer) { in GetListOfPassesAsString() argument
56 for (const auto& name : optimizer.GetPassNames()) { in GetListOfPassesAsString()
65 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetLegalizationPasses() local
66 optimizer.RegisterLegalizationPasses(); in GetLegalizationPasses()
67 return GetListOfPassesAsString(optimizer); in GetLegalizationPasses()
71 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetOptimizationPasses() local
72 optimizer.RegisterPerformancePasses(); in GetOptimizationPasses()
73 return GetListOfPassesAsString(optimizer); in GetOptimizationPasses()
77 spvtools::Optimizer optimizer(kDefaultEnvironment); in GetSizePasses() local
78 optimizer.RegisterSizePasses(); in GetSizePasses()
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | single_loss_example.py | 39 optimizer = optimizer_fn()
48 dataset_fn, loss_fn, optimizer, distribution, iterations_per_step)
54 def minimize_loss_example(optimizer, use_bias=False, use_callable_loss=True): argument
72 if strategy_test_lib.is_optimizer_v2_instance(optimizer):
73 return optimizer.minimize(loss_fn, lambda: layer.trainable_variables)
75 return optimizer.minimize(loss_fn)
77 return optimizer.minimize(loss_fn())
97 optimizer = optimizer_fn()
115 if strategy_test_lib.is_optimizer_v2_instance(optimizer):
116 return optimizer.minimize(loss_fn, lambda: layer.trainable_variables)
[all …]
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2_correctness_test.py | 139 optimizer = tpu_embedding_v2_utils.SGD(learning_rate=0.1)
141 optimizer = tpu_embedding_v2_utils.Adagrad(learning_rate=0.1)
143 optimizer = tpu_embedding_v2_utils.Adam(learning_rate=0.1)
146 mid_level_api = self._create_mid_level(optimizer=optimizer)
148 return strategy, mid_level_api, optimizer
156 strategy, mid_level_api, optimizer = (
207 optimizer)
209 def _create_mid_level(self, optimizer=None): argument
211 if optimizer is None:
212 optimizer = tpu_embedding_v2_utils.SGD(learning_rate=0.1)
[all …]
|
| D | tpu_embedding_v2_utils_test.py | 33 def test_grad_clip_with_accumulation_off(self, optimizer): argument
35 optimizer(use_gradient_accumulation=False, clipvalue=0.)
37 optimizer(use_gradient_accumulation=False, clipvalue=(None, 1.))
42 def test_grad_clip_with_tuple(self, optimizer): argument
43 opt = optimizer(clipvalue=(-1., 1.))
50 def test_grad_clip_with_single_value(self, optimizer): argument
51 opt = optimizer(clipvalue=1.)
58 def test_grad_clip_with_tuple_and_none(self, optimizer): argument
59 opt = optimizer(clipvalue=(None, 1))
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/keras_examples_benchmarks/ |
| D | mnist_conv_custom_training_benchmark_test.py | 67 def train_step(self, inputs, model, loss_fn, optimizer, batch_size): argument
85 optimizer.apply_gradients(zip(grads, model.trainable_weights))
89 def distributed_train_step(self, batch_dataset, model, loss_fn, optimizer, argument
111 optimizer,
121 optimizer, argument
160 loss_fn, optimizer,
165 optimizer, batch_size)
177 optimizer, argument
222 if not isinstance(optimizer, tf.keras.optimizers.Optimizer):
235 self.train_function(model, dataset, loss_fn, optimizer, 1,
[all …]
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | test_example.py | 31 def minimize_loss_example(optimizer, use_bias=False, use_callable_loss=True): argument
49 if isinstance(optimizer, optimizer_v2.OptimizerV2):
50 return optimizer.minimize(loss_fn, lambda: layer.trainable_variables)
52 return optimizer.minimize(loss_fn)
54 return optimizer.minimize(loss_fn())
74 optimizer = optimizer_fn()
92 if isinstance(optimizer, optimizer_v2.OptimizerV2):
93 return optimizer.minimize(loss_fn, lambda: layer.trainable_variables)
96 return optimizer.minimize(loss_fn)
|
| D | multi_worker_test.py | 54 if isinstance(model.optimizer, optimizer_v1.TFOptimizer):
55 optimizer = model.optimizer
58 while isinstance(optimizer, optimizer_v1.TFOptimizer):
59 optimizer = optimizer.optimizer
60 optimizer = copy.deepcopy(optimizer)
62 optimizer_config = model.optimizer.get_config()
63 optimizer = type(model.optimizer).from_config(optimizer_config)
66 optimizer,
|
| D | keras_utils_test.py | 86 optimizer='sgd',
140 optimizer='sgd',
164 optimizer='sgd',
245 optimizer = gradient_descent.GradientDescentOptimizer(0.001)
249 optimizer,
370 optimizer=optimizer_combinations
373 def test_masking(self, distribution, optimizer): argument
385 optimizer=optimizer())
402 optimizer=optimizer_combinations
404 def test_batchnorm_correctness(self, distribution, fused, optimizer): argument
[all …]
|
| D | simple_models.py | 54 optimizer = gradient_descent.SGD(learning_rate=0.001)
58 optimizer=optimizer)
79 optimizer = gradient_descent.SGD(learning_rate=0.001)
83 optimizer=optimizer)
109 optimizer = gradient_descent.SGD(learning_rate=0.001)
114 optimizer=optimizer)
|