Searched defs:value_fn (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_gather_test.py | 128 def value_fn(ctx): function
150 def value_fn(ctx): function
178 def value_fn(ctx): function
231 def value_fn(ctx): function
370 def value_fn(ctx): function
398 def value_fn(ctx): function
427 def value_fn(ctx): function
495 def value_fn(ctx): function
553 def value_fn(ctx): function
613 def value_fn(ctx): function
[all …]
|
| D | values_test.py | 127 def value_fn(ctx): function
147 def value_fn(ctx): function
167 def value_fn(ctx): function
190 def value_fn(ctx): function
215 def value_fn(ctx): function
238 def value_fn(ctx): function
259 def value_fn(ctx): function
289 def value_fn(ctx): function
311 def value_fn(ctx): function
|
| D | cross_device_ops_test.py | 242 value_fn = lambda device_idx: inputs[pid * len(devices) + device_idx] function
285 def value_fn(device_idx, idx=i): function
|
| D | collective_all_reduce_strategy.py | 624 def _experimental_distribute_values_from_function(self, value_fn): argument
|
| D | one_device_strategy.py | 341 def _experimental_distribute_values_from_function(self, value_fn): argument
|
| D | parameter_server_strategy.py | 401 def _experimental_distribute_values_from_function(self, value_fn): argument
|
| D | distribute_lib.py | 1623 def experimental_distribute_values_from_function(self, value_fn): argument
2216 def _experimental_distribute_values_from_function(self, value_fn): argument
3431 def _experimental_distribute_values_from_function(self, value_fn): argument
|
| D | mirrored_strategy.py | 591 def _experimental_distribute_values_from_function(self, value_fn): argument
|
| D | tpu_strategy.py | 979 def _experimental_distribute_values_from_function(self, value_fn): argument
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure_test.py | 85 def testFlatStructure(self, value_fn, expected_structure, expected_types, argument
330 def testRoundTripConversion(self, value_fn): argument
709 def testToBatchedTensorList(self, value_fn, element_0_fn): argument
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | ops_test.py | 397 def testCastToPrimitiveTypesFrom(self, value_fn): argument
|