| /external/tensorflow/tensorflow/python/training/ |
| D | training.py | 30 from tensorflow.python.training.adadelta import AdadeltaOptimizer
31 from tensorflow.python.training.adagrad import AdagradOptimizer
32 from tensorflow.python.training.adagrad_da import AdagradDAOptimizer
33 from tensorflow.python.training.proximal_adagrad import ProximalAdagradOptimizer
34 from tensorflow.python.training.adam import AdamOptimizer
35 from tensorflow.python.training.ftrl import FtrlOptimizer
36 from tensorflow.python.training.experimental.loss_scale_optimizer import MixedPrecisionLossScaleOpt…
37 from tensorflow.python.training.experimental.mixed_precision import enable_mixed_precision_graph_re…
38 from tensorflow.python.training.experimental.mixed_precision import enable_mixed_precision_graph_re…
39 from tensorflow.python.training.momentum import MomentumOptimizer
[all …]
|
| D | evaluation_test.py | 37 from tensorflow.python.training import basic_session_run_hooks
38 from tensorflow.python.training import evaluation
39 from tensorflow.python.training import gradient_descent
40 from tensorflow.python.training import monitored_session
41 from tensorflow.python.training import saver
42 from tensorflow.python.training import training
95 training.get_or_create_global_step())
144 single_input, single_label = training.slice_input_producer(
146 inputs, labels = training.batch([single_input, single_label], batch_size=6,
|
| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | resnet50.py | 73 def call(self, input_tensor, training=False): argument
75 x = self.bn2a(x, training=training)
79 x = self.bn2b(x, training=training)
83 x = self.bn2c(x, training=training)
148 def call(self, input_tensor, training=False): argument
150 x = self.bn2a(x, training=training)
154 x = self.bn2b(x, training=training)
158 x = self.bn2c(x, training=training)
161 shortcut = self.bn_shortcut(shortcut, training=training)
297 def call(self, inputs, training=True, intermediates_dict=None): argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | keras_imagenet_main_graph_mode.pbtxt | 4464 name: "training/SGD/iter"
4489 s: "training/SGD/iter"
4494 name: "training/SGD/bn2a_branch1/beta/momentum"
4522 s: "training/SGD/bn2a_branch1/beta/momentum"
4527 name: "training/SGD/bn2a_branch1/gamma/momentum"
4555 s: "training/SGD/bn2a_branch1/gamma/momentum"
4560 name: "training/SGD/bn2a_branch2a/beta/momentum"
4588 s: "training/SGD/bn2a_branch2a/beta/momentum"
4593 name: "training/SGD/bn2a_branch2a/gamma/momentum"
4621 s: "training/SGD/bn2a_branch2a/gamma/momentum"
[all …]
|
| D | keras_imagenet_main.pbtxt | 6751 name: "training/LossScaleOptimizer/truediv"
6798 …name: "training/LossScaleOptimizer/gradients/loss_1/res4c_branch2c/kernel/Regularizer/Square_grad/…
6986 …name: "training/LossScaleOptimizer/gradients/loss_1/res4c_branch2b/kernel/Regularizer/Square_grad/…
7140 …name: "training/LossScaleOptimizer/gradients/loss_1/res4c_branch2a/kernel/Regularizer/Square_grad/…
7340 …name: "training/LossScaleOptimizer/gradients/loss_1/res2a_branch2c/kernel/Regularizer/Square_grad/…
7845 …name: "training/LossScaleOptimizer/gradients/loss_1/res4a_branch1/kernel/Regularizer/Square_grad/M…
8165 …name: "training/LossScaleOptimizer/gradients/loss_1/res4a_branch2a/kernel/Regularizer/Square_grad/…
8356 …name: "training/LossScaleOptimizer/gradients/loss_1/res3d_branch2c/kernel/Regularizer/Square_grad/…
8510 …name: "training/LossScaleOptimizer/gradients/loss_1/res3d_branch2b/kernel/Regularizer/Square_grad/…
8664 …name: "training/LossScaleOptimizer/gradients/loss_1/res3d_branch2a/kernel/Regularizer/Square_grad/…
[all …]
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | ragged_keras_tensor_test.py | 31 from tensorflow.python.keras.engine import training
66 model = training.Model(inp, out)
74 model = training.Model(inp, out)
82 model = training.Model(inp, out)
90 model = training.Model(inp, out)
104 model = training.Model(inp, out)
112 model2 = training.Model.from_config(model_config)
158 model = training.Model(inp, out)
170 model2 = training.Model.from_config(model_config)
182 model = training.Model(inp, out)
[all …]
|
| D | training_eager_v1.py | 90 training=False): argument
117 kwargs['training'] = training
154 if training:
229 training=False): argument
252 with backend.eager_learning_phase_scope(1 if training else 0), \
262 training=training))
267 if training:
316 training=True,
359 training=False,
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | normalization_test.py | 38 from tensorflow.python.training import gradient_descent
39 from tensorflow.python.training import saver as saver_lib
55 training = not freeze_mode
56 bn = bn_layer.apply(conv, training=training)
265 training = array_ops.placeholder(dtype='bool')
266 outputs = bn.apply(inputs, training=training)
292 training = array_ops.placeholder(dtype='bool')
293 outputs = bn.apply(inputs, training=training)
322 training = array_ops.placeholder(dtype='bool')
323 outputs = bn.apply(inputs, training=training)
[all …]
|
| D | normalization.py | 171 def call(self, inputs, training=False): argument
172 return super(BatchNormalization, self).call(inputs, training=training)
190 training=False, argument
338 return layer.apply(inputs, training=training)
|
| D | core.py | 226 def call(self, inputs, training=False): argument
227 return super(Dropout, self).call(inputs, training=training)
235 training=False, argument
272 return layer.apply(inputs, training=training)
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | custom_training_loop_test.py | 63 def call(self, inputs, training=None): argument
64 self.training = training
65 if training:
102 y_pred = model(x, training=True)
106 return loss, model(x, training=False)
162 return model(x), model(x, training=False), model(x, training=True)
189 self.assertFalse(model.layer.training)
192 self.assertEqual(model.layer.training, 0)
195 self.assertEqual(model.layer.training, 1)
219 def call(self, inputs, training=False): argument
[all …]
|
| D | tracking_test.py | 32 from tensorflow.python.keras.engine import training
41 from tensorflow.python.training.tracking import base
42 from tensorflow.python.training.tracking import data_structures
43 from tensorflow.python.training.tracking import util
46 class HasList(training.Model):
117 model = training.Model()
120 model2 = training.Model()
126 class _Subclassed(training.Model):
158 class MyModel2(training.Model):
201 class HasEqualContainers(training.Model):
[all …]
|
| D | get_config_test.py | 23 from tensorflow.python.keras.engine import training
32 model = training.Model.from_config(get_config_samples.FUNCTIONAL_DNN)
36 model = training.Model.from_config(get_config_samples.FUNCTIONAL_CNN)
40 model = training.Model.from_config(get_config_samples.FUNCTIONAL_LSTM)
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | noise.py | 64 def call(self, inputs, training=None): argument
73 return K.in_train_phase(noised, inputs, training=training)
115 def call(self, inputs, training=None): argument
126 return K.in_train_phase(noised, inputs, training=training)
179 def call(self, inputs, training=None): argument
202 return K.in_train_phase(dropped_inputs, inputs, training=training)
|
| D | normalization.py | 538 def _fused_batch_norm(self, inputs, training): argument
618 training, train_op, _fused_batch_norm_inference)
621 training_value = control_flow_util.constant_value(training)
625 momentum = control_flow_util.smart_cond(training,
652 def _renorm_correction_and_moments(self, mean, variance, training, argument
680 r = control_flow_util.smart_cond(training, lambda: r,
682 d = control_flow_util.smart_cond(training, lambda: d,
698 return control_flow_util.smart_cond(training, _do_update, _fake_update)
728 def _get_training_value(self, training=None): argument
729 if training is None:
[all …]
|
| /external/tensorflow/tensorflow/python/estimator/ |
| D | training.py | 26 from tensorflow_estimator.python.estimator import training
30 training.__all__ = [s for s in dir(training) if not s.startswith('__')]
32 from tensorflow_estimator.python.estimator.training import *
|
| /external/rnnoise/ |
| D | TRAINING-README | 3 (2) ./denoise_training signal.raw noise.raw count > training.f32
7 (3) cd training ; ./bin2hdf5.py ../src/training.f32 500000 87 training.h5
|
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | utils.py | 158 training = get_training_arg(training_arg_index, args, kwargs)
159 if training is None:
160 training = default_training_value or K.learning_phase()
165 def replace_training_and_call(training): argument
166 set_training_arg(training, training_arg_index, args, kwargs)
170 training, lambda: replace_training_and_call(True),
224 def set_training_arg(training, index, args, kwargs): argument
228 args[index] = training
230 kwargs['training'] = training
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | BUILD | 122 "//tensorflow/python:training",
186 "//tensorflow/python:training",
189 "//tensorflow/python/training:saver_test_utils",
292 "//tensorflow/python/training/tracking:base",
355 "//tensorflow/python/training/saving:checkpoint_options",
356 "//tensorflow/python/training/saving:functional_saver",
357 "//tensorflow/python/training/saving:saveable_object_util",
358 "//tensorflow/python/training/tracking",
359 "//tensorflow/python/training/tracking:base",
360 "//tensorflow/python/training/tracking:graph_view",
[all …]
|
| /external/llvm-project/llvm/test/Transforms/Inline/ML/ |
| D | development-training-log.ll | 3 …l-inliner=development -passes=scc-oz-module-inliner -training-log=- -ml-inliner-model-under-traini…
4 …l-inliner=development -passes=scc-oz-module-inliner -training-log=- -ml-inliner-model-under-traini…
5 ; RUN: opt -enable-ml-inliner=development -passes=scc-oz-module-inliner -training-log=- -ml-inliner…
6 …l-inliner=development -passes=scc-oz-module-inliner -training-log=- -ml-inliner-model-under-traini…
7 ; RUN: opt -enable-ml-inliner=development -passes=scc-oz-module-inliner -training-log=- -S < %s | F…
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | function_test.py | 35 def call(self, inputs, training=True): argument
42 def call(self, inputs, training=True): argument
43 return super(DefunnedMiniModel, self).call(inputs, training=training)
103 instance_call_one = m.call(tf.ones([1, 2]), training=True)
105 inputs=tf.ones([1, 2]), training=True)
106 class_call = DefunnedMiniModel.call(m, tf.ones([1, 2]), training=True)
131 tf.ones([1, 2]), training=False)
133 inputs=tf.ones([1, 2]), training=False)
139 m, tf.ones([1, 2]), training=False)
141 m, inputs=tf.ones([1, 2]), training=True)
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | BUILD | 33 "//tensorflow/python/training/saving:saveable_object",
114 "//tensorflow/python/training:optimizer",
115 "//tensorflow/python/training/saving:saveable_object",
116 "//tensorflow/python/training/saving:saveable_object_util",
151 "//tensorflow/python/training/saving:checkpoint_options",
152 "//tensorflow/python/training/saving:functional_saver",
153 "//tensorflow/python/training/saving:saveable_object_util",
190 "//tensorflow/python/training/saving:checkpoint_options",
212 "//tensorflow/python:training",
253 target = "//tensorflow/python/training/tracking:benchmarks_test",
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | image_preprocessing.py | 226 def call(self, inputs, training=True): argument
227 if training is None:
228 training = K.learning_phase()
276 output = control_flow_util.smart_cond(training, random_cropped_inputs,
403 def call(self, inputs, training=True): argument
404 if training is None:
405 training = K.learning_phase()
417 output = control_flow_util.smart_cond(training, random_flipped_inputs,
531 def call(self, inputs, training=True): argument
532 if training is None:
[all …]
|
| /external/tensorflow/tensorflow/python/keras/feature_column/ |
| D | dense_features.py | 119 def call(self, features, cols_to_output_tensors=None, training=None): argument
155 if training is None:
156 training = backend.learning_phase()
166 transformation_cache, self._state_manager, training=training)
|
| D | sequence_feature_column.py | 119 def call(self, features, training=None): argument
147 if training is None:
148 training = backend.learning_phase()
157 transformation_cache, self._state_manager, training=training)
|