| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_gather_test.py | 126 def value_fn(ctx): function
148 def value_fn(ctx): function
176 def value_fn(ctx): function
224 def value_fn(ctx): function
363 def value_fn(ctx): function
391 def value_fn(ctx): function
420 def value_fn(ctx): function
488 def value_fn(ctx): function
541 def value_fn(ctx): function
601 def value_fn(ctx): function
[all …]
|
| D | values_test.py | 80 def value_fn(ctx): function
100 def value_fn(ctx): function
120 def value_fn(ctx): function
143 def value_fn(ctx): function
168 def value_fn(ctx): function
191 def value_fn(ctx): function
212 def value_fn(ctx): function
244 def value_fn(ctx): function
270 def value_fn(ctx): function
301 def value_fn(ctx): function
|
| D | cross_device_ops_test.py | 240 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 | 683 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 | 400 def _experimental_distribute_values_from_function(self, value_fn): argument
|
| D | distribute_lib.py | 1680 def experimental_distribute_values_from_function(self, value_fn): argument
2293 def _experimental_distribute_values_from_function(self, value_fn): argument
3660 def _experimental_distribute_values_from_function(self, value_fn): argument
|
| D | mirrored_strategy.py | 597 def _experimental_distribute_values_from_function(self, value_fn): argument
|
| D | tpu_strategy.py | 1037 def _experimental_distribute_values_from_function(self, value_fn): argument
|
| /external/tensorflow/tensorflow/core/lib/monitoring/ |
| D | collection_registry.h | 305 inline void CollectValue(std::function<int64_t()> value_fn, in CollectValue()
318 inline void CollectValue(std::function<std::string()> value_fn, in CollectValue()
331 inline void CollectValue(std::function<bool()> value_fn, Point* const point) { in CollectValue()
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure_test.py | 503 def testFlatStructure(self, value_fn, expected_structure_fn, argument
588 def testRoundTripConversion(self, value_fn): argument
890 def testToBatchedTensorList(self, value_fn, element_0_fn): argument
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | ops_test.py | 393 def testCastToPrimitiveTypesFrom(self, value_fn): argument
|
| /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()
|