| /external/tensorflow/tensorflow/contrib/eager/python/examples/linear_regression/ |
| linear_regression_graph_test.py | 53 optimization_step = tf.train.GradientDescentOptimizer(
59 def train(num_epochs): function in function:GraphLinearRegressionBenchmark.benchmarkGraphLinearRegression
69 train(1)
72 train(num_epochs)
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| learning.py | 31 optimizer = tf.train.MomentumOptimizer(FLAGS.learning_rate, FLAGS.momentum)
43 learning.train(train_op,
58 def train(train_op, function
83 """Wrapper around tf-slim's train function.
136 sync_optimizer: an instance of tf.train.SyncReplicasOptimizer, or a list of
162 total_loss, _ = _slim.learning.train(
|
| /external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/ |
| cifar10_train.py | 15 """A binary to train pruned CIFAR-10 using a single GPU.
33 data set, compile the program and train the model.
55 def train(): function
56 """Train CIFAR-10 for a number of steps."""
72 train_op = cifar10.train(loss, global_step)
90 class _LoggerHook(tf.train.SessionRunHook):
99 return tf.train.SessionRunArgs(loss) # Asks for loss value.
114 with tf.train.MonitoredTrainingSession(
116 hooks=[tf.train.StopAtStepHook(last_step=FLAGS.max_steps),
117 tf.train.NanTensorHook(loss)
[all...] |
| cifar10_pruning.py | 30 train_op = train(loss, global_step)
153 eval_data: bool, indicating if one should use the train or eval data set.
309 loss_averages = tf.train.ExponentialMovingAverage(0.9, name='avg')
324 def train(total_loss, global_step): function
325 """Train CIFAR-10 model.
342 lr = tf.train.exponential_decay(
355 opt = tf.train.GradientDescentOptimizer(lr)
371 variable_averages = tf.train.ExponentialMovingAverage(MOVING_AVERAGE_DECAY,
376 train_op = tf.no_op(name='train')
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| training.py | 36 optimizer = tf.train.MomentumOptimizer(FLAGS.learning_rate, FLAGS.momentum)
42 tf.contrib.training.train(train_op, my_log_dir)
48 In order to use the `train` function, one needs a train_op: an `Operation` that
144 tf.contrib.training.train(train_op, my_log_dir, scaffold=scaffold)
177 tf.contrib.training.train(train_op, my_log_dir, scaffold=scaffold)
208 tf.contrib.training.train(train_op, my_log_dir, scaffold=scaffold)
240 tf.contrib.training.train(train_op, my_log_dir, scaffold=scaffold)
267 'train',
393 variables_to_train: an optional list of variables to train. If None, it will
477 def train(train_op function
[all...] |
| /external/tensorflow/tensorflow/examples/tutorials/mnist/ |
| mnist_with_summaries.py | 38 def train(): function
127 with tf.name_scope('train'):
128 train_step = tf.train.AdamOptimizer(FLAGS.learning_rate).minimize(
141 train_writer = tf.summary.FileWriter(FLAGS.log_dir + '/train', sess.graph)
145 # Train the model, and also write summaries.
149 def feed_dict(train):
151 if train or FLAGS.fake_data:
152 xs, ys = mnist.train.next_batch(100, fake_data=FLAGS.fake_data)
164 else: # Record train set summaries, and train
[all...] |
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| learning.py | 22 tf.train.Supervisor and its managed_session in its implementation to ensure the
38 optimizer = tf.train.MomentumOptimizer(FLAGS.learning_rate, FLAGS.momentum)
44 slim.learning.train(train_op, my_log_dir)
50 In order to train, TF-Slim's train loop needs a train_op: an `Operation` that
143 slim.learning.train(train_op, my_log_dir, init_fn=InitAssignFn)
177 slim.learning.train(train_op, my_log_dir, init_fn=InitAssignFn)
210 slim.learning.train(train_op, my_log_dir, init_fn=InitAssignFn)
244 slim.learning.train(train_op, my_log_dir, init_fn=InitAssignFn)
276 'create_train_op', 'train_step', 'train'
531 def train(train_op, function
[all...] |
| /external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/ |
| rnn_ptb.py | 188 def train(model, optimizer, train_data, sequence_length, clip_ratio): function
225 self.train = self.tokenize(os.path.join(path, "ptb.train.txt"))
298 train_data = _divide_into_batches(corpus.train, FLAGS.batch_size)
305 tf.train.latest_checkpoint(FLAGS.logdir)):
314 optimizer = tf.train.GradientDescentOptimizer(learning_rate)
318 train(model, optimizer, train_data, FLAGS.seq_len, FLAGS.clip)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| graph_actions.py | 61 'graph_actions.py will be deleted. Use tf.train.* utilities instead. '
123 def train(graph, function
142 """Train a model.
158 graph: A graph to train. It is expected that this graph is not in use
180 arg to tf.train.Saver constructor.
190 max_steps: Number of total steps for which to train model. If `None`,
191 train forever. Two calls fit(steps=100) means 200 training iterations.
251 """See train."""
348 'Global step was not incremented by train op at step %s'
496 graph: A `Graph` to train. It is expected that this graph is not in us
[all...] |
| experiment.py | 119 """Experiment is a class containing all information needed to train a model.
176 means train forever.
200 Perform this many (integer) number of train steps for each
246 "Please call `TPUEstimator` train/evaluate directly. \n"
329 def train(self, delay_secs=None): member in class:Experiment
332 Train the estimator for `self._train_steps` steps, after waiting for
333 `delay_secs` seconds. If `self._train_steps` is `None`, train forever.
346 # to train. We might as well start as soon as we can.
503 # Exit if we have already reached number of steps to train.
658 self.train(delay_secs=0
[all...] |
| experiment_test.py | 112 def train(self, **kwargs): member in class:TestBaseEstimator
151 return super(TestEstimator, self).train(**kwargs)
153 def train(self, **kwargs): member in class:TestEstimator
154 raise ValueError('`train` is not defined in Estimator.')
174 def train(self, **kwargs): member in class:TestCoreEstimator
175 _check_method_supports_args(core_estimator.Estimator.train, kwargs)
178 return super(TestCoreEstimator, self).train(**kwargs)
248 fit_args = ex.train(delay_secs=0)
261 ex.train(delay_secs=delay)
277 ex.train()
[all...] |
| learn_runner_test.py | 90 def train(self): member in class:TestExperiment
91 return "train-{}".format(self._model_dir)
342 "train-" + _MODIR_DIR,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| gbdt_batch.py | 233 mode: Train/Eval/Infer
239 if mode == learn.ModeKeys.TRAIN:
355 mode: learn.ModeKeys.TRAIN or EVAL or INFER.
382 apply_averaging=mode != learn.ModeKeys.TRAIN,
404 mode: Mode the graph is running in (train|predict|eval).
468 def train(self, loss, predictions_dict, labels): member in class:GradientBoostedDecisionTreeModel
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| sdca_ops_test.cc | 233 Graph* train = nullptr; local
236 20 /* dense features per group */, &init, &train);
238 test::Benchmark("cpu", train, GetSingleThreadedOptions(), init).Run(iters);
244 Graph* train = nullptr; local
247 200000 /* dense features per group */, &init, &train);
249 test::Benchmark("cpu", train, GetSingleThreadedOptions(), init).Run(iters);
255 Graph* train = nullptr; local
258 0 /* dense features per group */, &init, &train);
260 test::Benchmark("cpu", train, GetMultiThreadedOptions(), init).Run(iters);
|
| training_ops_test.cc | 82 Graph* train; local
83 SGD(params, &init, &train);
84 test::Benchmark("cpu", train, GetOptions(), init).Run(iters);
114 Graph* train; local
115 Adagrad(params, &init, &train);
116 test::Benchmark("cpu", train, GetOptions(), init).Run(iters);
148 Graph* train; local
149 Momentum(params, &init, &train);
150 test::Benchmark("cpu", train, GetOptions(), init).Run(iters);
191 Graph* train; local
231 Graph* train; local
268 Graph* train; local
305 Graph* train; local
[all...] |
| /frameworks/base/core/java/android/service/resolver/ |
| ResolverRankerService.java | 95 * Called by the system to train/update a ranking service, after the user makes a selection from
172 public void train(final List<ResolverTarget> targets, final int selectedPosition) method in class:ResolverRankerService.ResolverRankerServiceWrapper
179 Log.d(TAG, "train calls onTranRankingModel");
183 Log.e(TAG, "onTrainRankingModel failed; skip train: " + e);
|
| /external/tensorflow/tensorflow/python/estimator/ |
| estimator.py | 69 """Estimator class to train and evaluate TensorFlow models.
125 passed to `train`, `evaluate`, and `predict`. This should be a
128 passed to `train`, `evaluate`, and `predict`. This should be a
286 def train(self, member in class:Estimator
310 steps: Number of steps for which to train model. If `None`, train forever
311 or train until input_fn generates the `OutOfRange` error or
313 times train(steps=10) then training occurs in total 20 steps. If
317 max_steps: Number of total steps for which to train model. If `None`,
318 train forever or train until input_fn generates the `OutOfRange` erro
[all...] |
| /external/opencv/ml/src/ |
| mlknearest.cpp | 65 train( _train_data, _responses, _sample_idx, _is_regression, _max_k, false );
92 bool CvKNearest::train( const CvMat* _train_data, const CvMat* _responses, function in class:CvKNearest
99 CV_FUNCNAME( "CvKNearest::train" );
113 CV_CALL( cvPrepareTrainData( "CvKNearest::train", _train_data, CV_ROW_SAMPLE,
320 CV_ERROR( CV_StsError, "The search tree must be constructed first using train method" );
|
| mlnbayes.cpp | 103 train( _train_data, _responses, _var_idx, _sample_idx );
107 bool CvNormalBayesClassifier::train( const CvMat* _train_data, const CvMat* _responses, function in class:CvNormalBayesClassifier
118 CV_FUNCNAME( "CvNormalBayesClassifier::train" );
195 /* process train data (count, sum , productsum) */
|
| mlrtrees.cpp | 55 bool CvForestTree::train( CvDTreeTrainData* _data, function in class:CvForestTree
61 CV_FUNCNAME( "CvForestTree::train" );
80 CvForestTree::train( const CvMat*, int, const CvMat*, const CvMat*, function in class:CvForestTree
89 CvForestTree::train( CvDTreeTrainData*, const CvMat* ) function in class:CvForestTree
231 bool CvRTrees::train( const CvMat* _train_data, int _tflag, function in class:CvRTrees
238 CV_FUNCNAME("CvRTrees::train");
374 CV_CALL(tree->train( data, sample_idx_for_tree, this ));
|
| mlem.cpp | 84 // just invoke the train() method
85 train(samples, sample_idx, params, labels);
303 bool CvEM::train( const CvMat* _samples, const CvMat* _sample_idx, function in class:CvEM
[all...] |
| mlann_mlp.cpp | 816 int CvANN_MLP::train( const CvMat* _inputs, const CvMat* _outputs, function in class:CvANN_MLP
829 CV_FUNCNAME( "CvANN_MLP::train" );
1059 CV_FUNCNAME( "CvANN_MLP::train" );
[all...] |
| mlboost.cpp | 103 CvBoostTree::train( CvDTreeTrainData* _train_data, function in class:CvBoostTree
116 CvBoostTree::train( const CvMat*, int, const CvMat*, const CvMat*, function in class:CvBoostTree
125 CvBoostTree::train( CvDTreeTrainData*, const CvMat* ) function in class:CvBoostTree
864 train( _train_data, _tflag, _responses, _var_idx, _sample_idx,
908 CvBoost::train( const CvMat* _train_data, int _tflag, function in class:CvBoost
917 CV_FUNCNAME( "CvBoost::train" );
949 if( !tree->train( data, subsample_mask, this ) )
[all...] |
| mlsvm.cpp | 1185 train( _train_data, _responses, _var_idx, _sample_idx, _params );
1525 bool CvSVM::train( const CvMat* _train_data, const CvMat* _responses, function in class:CvSVM
[all...] |
| /frameworks/base/core/java/com/android/internal/app/ |
| ResolverComparator.java | 389 mRanker.train(mTargets, selectedPos);
396 Log.e(TAG, "Error in Train: " + e);
|
