| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_gather_test.py | 126 def value_fn(ctx): function
131 value_fn)
148 def value_fn(ctx): function
153 value_fn)
176 def value_fn(ctx): function
181 value_fn)
224 def value_fn(ctx): function
228 value_fn)
363 def value_fn(ctx): function
368 value_fn)
[all …]
|
| D | values_test.py | 80 def value_fn(ctx): function
85 distribution.experimental_distribute_values_from_function(value_fn))
100 def value_fn(ctx): function
105 distribution.experimental_distribute_values_from_function(value_fn))
120 def value_fn(ctx): function
124 distribution.experimental_distribute_values_from_function(value_fn))
143 def value_fn(ctx): function
149 distribution.experimental_distribute_values_from_function(value_fn))
168 def value_fn(ctx): function
174 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 | 1680 def experimental_distribute_values_from_function(self, value_fn): argument
1754 value_fn)
2293 def _experimental_distribute_values_from_function(self, value_fn): argument
3660 def _experimental_distribute_values_from_function(self, value_fn): argument
3661 return value_fn(ValueContext())
|
| D | cross_device_ops_test.py | 240 value_fn = lambda device_idx: inputs[pid * len(devices) + device_idx] function
241 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 | 400 def _experimental_distribute_values_from_function(self, value_fn): argument
404 value_fn(distribute_lib.ValueContext(replica_id,
|
| D | collective_all_reduce_strategy.py | 683 def _experimental_distribute_values_from_function(self, value_fn): argument
691 per_replica_values.append(value_fn(value_context))
|
| D | mirrored_strategy.py | 597 def _experimental_distribute_values_from_function(self, value_fn): argument
600 per_replica_values.append(value_fn(
|
| D | tpu_strategy.py | 1037 def _experimental_distribute_values_from_function(self, value_fn): argument
1041 value_fn(distribute_lib.ValueContext(replica_id,
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure_test.py | 87 name, value_fn = y[:2]
90 value_fn=combinations.NamedObject("value_fn.{}".format(name), value_fn),
330 name, value_fn = y
332 value_fn=combinations.NamedObject("value_fn.{}".format(name), value_fn))
487 name, value_fn, element_0_fn = y
489 value_fn=combinations.NamedObject("value_fn.{}".format(name), value_fn),
503 def testFlatStructure(self, value_fn, expected_structure_fn, argument
505 value = value_fn()
557 for value_fn in not_equal_value_fns:
558 s3 = structure.type_spec_from_value(value_fn())
[all …]
|
| /external/tensorflow/tensorflow/core/lib/monitoring/ |
| D | collection_registry.h | 305 inline void CollectValue(std::function<int64_t()> value_fn, in CollectValue() argument
308 point->int64_value = value_fn(); in CollectValue()
318 inline void CollectValue(std::function<std::string()> value_fn, in CollectValue() argument
321 point->string_value = value_fn(); in CollectValue()
331 inline void CollectValue(std::function<bool()> value_fn, Point* const point) { in CollectValue() argument
333 point->bool_value = value_fn(); in CollectValue()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | optional_test.py | 68 name, value_fn, expected_structure = y
70 tf_value_fn=combinations.NamedObject(name, value_fn),
109 name, value, value_fn, gpu_compatible = y
112 tf_value_fn=combinations.NamedObject(name, value_fn),
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_simplify_padding_test.cc | 109 absl::FunctionRef<NativeT(absl::Span<const int64_t>, NativeT)> value_fn) { in SetConstantValue() argument
111 new_literal.MutableEachCell<int8_t>(value_fn); in SetConstantValue()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-multi-worker-mirrored-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.experimental.-central-storage-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.-parameter-server-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.-strategy.pbtxt | 44 argspec: "args=[\'self\', \'value_fn\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-mirrored-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 | 393 def testCastToPrimitiveTypesFrom(self, value_fn): argument
394 x = value_fn()
|