Home
last modified time | relevance | path

Searched refs:num_steps (Results 1 – 25 of 54) sorted by relevance

123

/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/
Dperiodic_test.py32 num_steps = 3
40 test_case=self, model=cycle, num_steps=num_steps)
43 num_steps = 3
53 test_case=self, model=cycle, num_steps=num_steps)
56 num_steps = 3
64 test_case=self, model=cycle, num_steps=num_steps)
67 num_steps = 3
77 test_case=self, model=cycle, num_steps=num_steps)
Dlevel_trend.py95 def transition_power_noise_accumulator(self, num_steps): argument
100 array_ops.concat(values=[array_ops.shape(num_steps), [2, 2]], axis=0))
102 num_steps = math_ops.cast(num_steps, self.dtype)
103 noise_transitions = num_steps - 1
119 num_steps * a + sum_of_first_n * (b + c) + sum_of_first_n_squares * d,
120 num_steps * b + sum_of_first_n * d,
121 num_steps * c + sum_of_first_n * d,
122 num_steps * d)
Dlevel_trend_test.py32 num_steps = 3
38 test_case=self, model=adder, num_steps=num_steps)
41 num_steps = 3
49 test_case=self, model=adder, num_steps=num_steps)
Dperiodic.py105 self, num_steps, noise_addition_coefficient=1): argument
141 math_ops.range(self._periodicity, dtype=num_steps.dtype),
144 array_ops.ones([array_ops.rank(num_steps)], dtype=dtypes.int32),
149 - (num_steps[..., None] - 1))
154 (num_steps[..., None] + reversed_remaining_steps) //
334 def transition_power_noise_accumulator(self, num_steps): argument
372 num_steps=ops.convert_to_tensor(
373 self._num_latent_values, dtype=num_steps.dtype))
375 num_steps.dtype)
376 full_periods = math_ops.cast(num_steps // periodicity_integer, self.dtype)
[all …]
Dtest_utils.py34 def transition_power_test_template(test_case, model, num_steps): argument
48 state_dimension, batch_shape=array_ops.shape(num_steps)).eval()
50 for iteration_number in range(num_steps):
61 def noise_accumulator_test_template(test_case, model, num_steps): argument
94 times=math_ops.range(num_steps + 1)[..., None], state=(None, None, 0))
96 num_steps=step_number)
97 for iteration_number in range(num_steps):
Dstate_space_model.py255 def transition_power_noise_accumulator(self, num_steps): argument
287 math_ops.reduce_max(num_steps) + 1,
290 return array_ops.gather(noise_additions, indices=num_steps)
346 def _cached_transition_powers_and_sums(self, num_steps): argument
349 indices=num_steps)
1121 def transition_power_noise_accumulator(self, num_steps): argument
1123 member_fn=lambda m: m.transition_power_noise_accumulator(num_steps),
1193 def transition_power_noise_accumulator(self, num_steps): argument
1199 num_steps=num_steps)
1205 self).transition_power_noise_accumulator(num_steps=num_steps)
/external/eigen/Eigen/src/Core/functors/
DNullaryFunctors.h45 linspaced_op_impl(const Scalar& low, const Scalar& high, Index num_steps) :
46 …w(low), m_high(high), m_size1(num_steps==1 ? 1 : num_steps-1), m_step(num_steps==1 ? Scalar() : (h…
92 linspaced_op_impl(const Scalar& low, const Scalar& high, Index num_steps) :
94 m_multiplier((high-low)/convert_index<Scalar>(num_steps<=1 ? 1 : num_steps-1)),
95 …m_divisor(convert_index<Scalar>((high>=low?num_steps:-num_steps)+(high-low))/((numext::abs(high-lo…
96 m_use_divisor(num_steps>1 && (numext::abs(high-low)+1)<num_steps)
131 linspaced_op(const Scalar& low, const Scalar& high, Index num_steps)
132 : impl((num_steps==1 ? high : low),high,num_steps)
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
Dautoregressive.py123 num_steps=None, argument
163 if num_steps is None:
164 num_steps = self._distribution0.event_shape.num_elements()
165 if num_steps is None:
168 if num_steps < 1:
169 raise ValueError("num_steps ({}) must be at least 1.".format(num_steps))
170 self._num_steps = num_steps
189 def num_steps(self): member in Autoregressive
/external/tensorflow/tensorflow/python/training/
Devaluation_test.py75 def _train_model(self, checkpoint_dir, num_steps): argument
99 hooks=[basic_session_run_hooks.StopAtStepHook(num_steps)]) as session:
104 if num_steps >= 300:
112 self._train_model(checkpoint_dir, num_steps=300)
139 self._train_model(checkpoint_dir, num_steps=300)
178 self._train_model(checkpoint_dir, num_steps=1)
206 self._train_model(checkpoint_dir, num_steps=1)
235 self._train_model(checkpoint_dir, num_steps=1)
Dbasic_session_run_hooks.py315 def __init__(self, num_steps=None, last_step=None, steps_per_run=1): argument
339 if num_steps is None and last_step is None:
341 if num_steps is not None and last_step is not None:
345 self._num_steps = num_steps
380 def __init__(self, num_steps=None, last_step=None): argument
399 if num_steps is None and last_step is None:
401 if num_steps is not None and last_step is not None:
403 self._num_steps = num_steps
/external/tensorflow/tensorflow/contrib/training/python/training/
Devaluation_test.py157 def _train_model(self, checkpoint_dir, num_steps): argument
181 hooks=[basic_session_run_hooks.StopAtStepHook(num_steps)])
183 if num_steps >= 300:
191 self._train_model(checkpoint_dir, num_steps=300)
217 self._train_model(checkpoint_dir, num_steps=1)
244 self._train_model(checkpoint_dir, num_steps=1)
276 def _train_model(self, checkpoint_dir, num_steps): argument
300 hooks=[basic_session_run_hooks.StopAtStepHook(num_steps)])
307 self._train_model(checkpoint_dir, num_steps=300)
362 self._train_model(checkpoint_dir, num_steps=300)
[all …]
Dtraining_test.py253 hooks=[basic_session_run_hooks.StopAtStepHook(num_steps=300)],
287 hooks=[basic_session_run_hooks.StopAtStepHook(num_steps=300)],
310 hooks=[basic_session_run_hooks.StopAtStepHook(num_steps=300)],
341 num_steps=number_of_steps[i]),
394 basic_session_run_hooks.StopAtStepHook(num_steps=1),
411 basic_session_run_hooks.StopAtStepHook(num_steps=300),
437 hooks=[basic_session_run_hooks.StopAtStepHook(num_steps=1)],
474 basic_session_run_hooks.StopAtStepHook(num_steps=200),
498 basic_session_run_hooks.StopAtStepHook(num_steps=300),
517 basic_session_run_hooks.StopAtStepHook(num_steps=400),
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dround_weights.cc35 int32 num_steps; in RoundWeights() local
37 context.GetOneInt32Parameter("num_steps", 256, &num_steps)); in RoundWeights()
40 [num_steps](const NodeMatch& match, const std::set<string>& input_nodes, in RoundWeights()
92 const float bucket_width = (max - min) / num_steps; in RoundWeights()
/external/tensorflow/tensorflow/contrib/distribute/python/
Dmonitor.py58 def run_steps(self, num_steps=None): argument
60 while num_steps is None or step < num_steps:
Destimator_integration_test.py108 num_steps = 10
112 training.TrainSpec(train_input_fn, max_steps=num_steps),
115 estimator.train(train_input_fn, steps=num_steps)
118 self.assertEqual(num_steps, scores[ops.GraphKeys.GLOBAL_STEP])
/external/tensorflow/tensorflow/python/keras/layers/
Dlocal_test.py39 num_steps = 8
67 input_shape=(num_samples, num_steps, input_dim))
71 num_steps = 8
96 layer.build((num_samples, num_steps, input_dim))
100 num_steps,
114 layer.build((num_samples, num_steps, input_dim))
/external/tensorflow/tensorflow/examples/tf2_showcase/
Dmnist.py141 num_steps=None): argument
145 if num_steps is not None and batch > num_steps:
222 flags_obj.log_interval, num_steps=1)
/external/tensorflow/tensorflow/core/grappler/clusters/
Dcluster.cc40 void Cluster::SetNumWarmupSteps(int num_steps) { in SetNumWarmupSteps() argument
42 num_steps); in SetNumWarmupSteps()
Dcluster.h73 void SetNumWarmupSteps(int num_steps);
/external/autotest/server/site_tests/firmware_ECThermal/
Dfirmware_ECThermal.py99 num_steps = len(self._thermal_setting[0]) - 3
102 ["\d+ K:\s+([0-9-]+) RPM"] * num_steps)
108 for i in xrange(num_steps + 1):
/external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
Dstargan_estimator_test.py231 num_steps = 10
232 est.train(train_input_fn, steps=num_steps)
236 self.assertEqual(num_steps, scores[ops.GraphKeys.GLOBAL_STEP])
/external/tensorflow/tensorflow/examples/tutorials/word2vec/
Dword2vec_basic.py227 num_steps = 100001
238 for step in xrange(num_steps):
259 if step == (num_steps - 1):
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.estimator.-stop-at-step-hook.pbtxt8 …argspec: "args=[\'self\', \'num_steps\', \'last_step\'], varargs=None, keywords=None, defaults=[\'…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.estimator.-stop-at-step-hook.pbtxt8 …argspec: "args=[\'self\', \'num_steps\', \'last_step\'], varargs=None, keywords=None, defaults=[\'…
Dtensorflow.train.-stop-at-step-hook.pbtxt8 …argspec: "args=[\'self\', \'num_steps\', \'last_step\'], varargs=None, keywords=None, defaults=[\'…

123