| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_gather_test.py | 128 def value_fn(ctx): function
133 value_fn)
150 def value_fn(ctx): function
155 value_fn)
178 def value_fn(ctx): function
183 value_fn)
231 def value_fn(ctx): function
235 value_fn)
370 def value_fn(ctx): function
375 value_fn)
[all …]
|
| D | values_test.py | 127 def value_fn(ctx): function
132 distribution.experimental_distribute_values_from_function(value_fn))
147 def value_fn(ctx): function
152 distribution.experimental_distribute_values_from_function(value_fn))
167 def value_fn(ctx): function
171 distribution.experimental_distribute_values_from_function(value_fn))
190 def value_fn(ctx): function
196 distribution.experimental_distribute_values_from_function(value_fn))
215 def value_fn(ctx): function
221 distribution.experimental_distribute_values_from_function(value_fn))
[all …]
|
| D | one_device_strategy.py | 341 def _experimental_distribute_values_from_function(self, value_fn): argument
345 return value_fn(distribute_lib.ValueContext())
|
| D | distribute_lib.py | 1623 def experimental_distribute_values_from_function(self, value_fn): argument
1697 value_fn)
2216 def _experimental_distribute_values_from_function(self, value_fn): argument
3431 def _experimental_distribute_values_from_function(self, value_fn): argument
3432 return value_fn(ValueContext())
|
| D | cross_device_ops_test.py | 242 value_fn = lambda device_idx: inputs[pid * len(devices) + device_idx] function
243 per_replica_value = make_per_replica_value(value_fn, devices)
285 def value_fn(device_idx, idx=i): function
288 per_replica_value = make_per_replica_value(value_fn, devices)
|
| D | parameter_server_strategy.py | 401 def _experimental_distribute_values_from_function(self, value_fn): argument
405 value_fn(distribute_lib.ValueContext(replica_id,
|
| D | collective_all_reduce_strategy.py | 624 def _experimental_distribute_values_from_function(self, value_fn): argument
632 per_replica_values.append(value_fn(value_context))
|
| D | mirrored_strategy.py | 591 def _experimental_distribute_values_from_function(self, value_fn): argument
594 per_replica_values.append(value_fn(
|
| D | tpu_strategy.py | 979 def _experimental_distribute_values_from_function(self, value_fn): argument
983 value_fn(distribute_lib.ValueContext(replica_id,
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure_test.py | 85 def testFlatStructure(self, value_fn, expected_structure, expected_types, argument
87 value = value_fn()
237 for value_fn in not_equal_value_fns:
238 s3 = structure.type_spec_from_value(value_fn())
330 def testRoundTripConversion(self, value_fn): argument
331 value = value_fn()
709 def testToBatchedTensorList(self, value_fn, element_0_fn): argument
710 batched_value = value_fn()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | optional_test.py | 72 name, value_fn, expected_structure = y
74 tf_value_fn=combinations.NamedObject(name, value_fn),
113 name, value, value_fn, gpu_compatible = y
116 tf_value_fn=combinations.NamedObject(name, value_fn),
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 45 argspec: "args=[\'self\', \'value_fn\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 46 argspec: "args=[\'self\', \'value_fn\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-multi-worker-mirrored-strategy.pbtxt | 45 argspec: "args=[\'self\', \'value_fn\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 45 argspec: "args=[\'self\', \'value_fn\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 45 argspec: "args=[\'self\', \'value_fn\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 45 argspec: "args=[\'self\', \'value_fn\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-strategy.pbtxt | 44 argspec: "args=[\'self\', \'value_fn\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-central-storage-strategy.pbtxt | 45 argspec: "args=[\'self\', \'value_fn\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-t-p-u-strategy.pbtxt | 49 argspec: "args=[\'self\', \'value_fn\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | ops_test.py | 397 def testCastToPrimitiveTypesFrom(self, value_fn): argument
398 x = value_fn()
|