Searched refs:experimental_local_results (Results 1 – 25 of 55) sorted by relevance
123
73 self.evaluate(distribution.experimental_local_results(assign))99 self.evaluate(distribution.experimental_local_results(assign_op))183 self.evaluate(distribution.experimental_local_results(assign))220 self.evaluate(distribution.experimental_local_results(ema_w))[0],241 self.evaluate(distribution.experimental_local_results(avg))[0],268 self.evaluate(distribution.experimental_local_results(w_apply))269 self.evaluate(distribution.experimental_local_results(w_assign))270 self.evaluate(distribution.experimental_local_results(w_apply))272 self.evaluate(distribution.experimental_local_results(ema_w))[0],281 self.evaluate(distribution.experimental_local_results(w_apply))[all …]
92 outputs = distribution.experimental_local_results(135 output = distribution.experimental_local_results(154 return distribution.experimental_local_results(215 output = distribution.experimental_local_results(233 return distribution.experimental_local_results(266 return nest.map_structure(distribution.experimental_local_results,288 return distribution.experimental_local_results(358 output = distribution.experimental_local_results(394 distribution.experimental_local_results(input_iterator.get_next()),410 return distribution.experimental_local_results(iterator.get_next())[all …]
97 self.assertEqual(d, strategy.experimental_local_results(var)[i].device)100 …strategy.experimental_local_results(var)[i]._distribute_strategy) # pylint: disable=protected-acc…131 return distribution.experimental_local_results(308 2.0, self.evaluate(distribution.experimental_local_results(v0)[0]))310 2.0, self.evaluate(distribution.experimental_local_results(v0)[1]))314 3.0, self.evaluate(distribution.experimental_local_results(v1)[0]))316 3.0, self.evaluate(distribution.experimental_local_results(v1)[1]))333 local_results = self.evaluate(distribution.experimental_local_results(v0))345 distribution.experimental_local_results(v1))364 local_results = self.evaluate(distribution.experimental_local_results(v0))[all …]
361 result = distribution.experimental_local_results(result)383 unwrapped = distribution.experimental_local_results(in_model_fn)391 unwrapped = distribution.experimental_local_results(in_model_fn)406 self.evaluate(distribution.experimental_local_results(result)))424 self.evaluate(distribution.experimental_local_results(result)))524 v0, v1 = distribution.experimental_local_results(v)541 v0, v1 = distribution.experimental_local_results(v)569 a0, a1 = distribution.experimental_local_results(a)570 b0, b1 = distribution.experimental_local_results(result_b)571 c0, c1 = distribution.experimental_local_results(result_c)[all …]
95 return distribution.experimental_local_results(141 return distribution.experimental_local_results(192 return distribution.experimental_local_results(241 distribution.experimental_local_results(266 distribution.experimental_local_results(339 per_replica_results = self.evaluate(distribution.experimental_local_results(371 distribution.experimental_local_results(distribution.run(assign)))389 distribution.experimental_local_results(distribution.run(add)))476 distribution.experimental_local_results(497 distribution.experimental_local_results([all …]
218 outputs = strategy.experimental_local_results(301 return strategy.experimental_local_results(430 outputs = strategy.experimental_local_results(train_step())614 strategy.experimental_local_results(train_step(next(input_iterator))))644 outputs2 = strategy.experimental_local_results(664 outputs = strategy.experimental_local_results(685 outputs = strategy.experimental_local_results(715 strategy.experimental_local_results,765 strategy.experimental_local_results,815 return strategy.experimental_local_results([all …]
49 values = strat.experimental_local_results(values)88 return [v.read_value() for v in strat.experimental_local_results(dv)]127 values = strat.experimental_local_results(results)156 values = strat.experimental_local_results(results)187 return strat.experimental_local_results(result)
89 output = distribution.experimental_local_results(112 return distribution.experimental_local_results(145 grads = distribution.experimental_local_results(train_step())
290 result = strategy.experimental_local_results(358 result = strategy.experimental_local_results(389 result = strategy.experimental_local_results(417 result = strategy.experimental_local_results(457 strategy.experimental_local_results(result[0]))459 strategy.experimental_local_results(result[1]))481 strategy.experimental_local_results(gathered_result))619 result = strategy.experimental_local_results(652 result = strategy.experimental_local_results(
350 list(strategy.experimental_local_results(next(iterator))))354 self.evaluate(strategy.experimental_local_results(next(iterator)))361 list(strategy.experimental_local_results(next(iterator))))428 global_step_tensors = strategy.experimental_local_results(value)457 x, y = self.evaluate((strategy.experimental_local_results(x),458 strategy.experimental_local_results(y)))534 map(strategy.experimental_local_results,556 strategy.experimental_local_results(576 strategy.experimental_local_results(673 map(strategy.experimental_local_results,[all …]
79 strategy.experimental_local_results(return_one()))83 strategy.experimental_local_results(return_two()))
115 def experimental_local_results(self, value): # pylint: disable=useless-super-delegation member in CentralStorageStrategy129 return super(CentralStorageStrategy, self).experimental_local_results(value)
138 return strategy.experimental_local_results(strategy.run(replica_fn))159 return strategy.experimental_local_results(strategy.run(replica_fn))213 return strategy.experimental_local_results(iterator.get_next())
59 distribution.experimental_local_results(
156 def experimental_local_results(self, value): # pylint: disable=useless-super-delegation member in OneDeviceStrategy170 return super(OneDeviceStrategy, self).experimental_local_results(value)
564 self.assertAllEqual(distribution.experimental_local_results(element), [i])835 distribution.experimental_local_results(x) for x in dist_dataset)837 distribution.experimental_local_results(x) for x in dist_dataset)866 distribution.experimental_local_results(x) for x in dist_dataset)868 distribution.experimental_local_results(x) for x in dist_dataset)900 data = nest.map_structure(distribution.experimental_local_results, data)1697 local_results = distribution.experimental_local_results(element)
151 strategy.experimental_local_results(train_step()),171 strategy.experimental_local_results(train_step()),192 strategy.experimental_local_results(train_step()),220 strategy.experimental_local_results(train_step()),247 strategy.experimental_local_results(train_step()),276 strategy.experimental_local_results(train_step()),299 strategy.experimental_local_results(train_step()),324 strategy.experimental_local_results(train_step()),349 strategy.experimental_local_results(step()),384 strategy.experimental_local_results(step(take_true_branch)),[all …]
101 self.evaluate(strategy.experimental_local_results(m.v)), [0.5, 0.5])253 strategy.experimental_local_results(strategy.run(loaded))),303 self.evaluate(strategy.experimental_local_results(m.v)), [3., 7.])314 strategy.experimental_local_results(strategy.run(loaded))),357 self.evaluate(strategy.experimental_local_results(m.v)), [5., 5.])369 self.evaluate(strategy.experimental_local_results(loaded.v)),373 self.evaluate(strategy.experimental_local_results(loaded.v)),398 strategy.experimental_local_results(strategy.run(loaded))),437 self.evaluate(strategy.experimental_local_results(loaded.v)), [0., 0.])440 self.evaluate(strategy.experimental_local_results(loaded.v)), [1., 1.])[all …]
102 variables = strategy.experimental_local_results(variable)119 variables = strategy.experimental_local_results(variable)142 accessed.extend(strategy.experimental_local_results(variable))
101 self.assertAllDifferent(distribution.experimental_local_results(kernel))103 self.assertAllDifferent(distribution.experimental_local_results(kernel))
69 return distribution.experimental_local_results(96 return distribution.experimental_local_results(
87 return nest.map_structure(distribution.experimental_local_results,112 return nest.map_structure(distribution.experimental_local_results,150 return nest.map_structure(distribution.experimental_local_results,254 return distribution.experimental_local_results(outputs)387 return nest.map_structure(distribution.experimental_local_results,491 return nest.map_structure(distribution.experimental_local_results,
260 fetches = distribution.experimental_local_results(479 not_reduced = distribution.experimental_local_results(528 self.assertLen(distribution.experimental_local_results(loss_output),533 unwrapped_output = distribution.experimental_local_results(loss_output)
98 updates = distribution.experimental_local_results(
51 out = strategy.experimental_local_results(