Searched refs:eager_mode (Results 1 – 25 of 74) sorted by relevance
123
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | summary_ops_test.py | 54 with context.eager_mode():
67 with context.eager_mode():
86 with context.eager_mode():
107 with context.eager_mode():
116 with context.eager_mode():
127 with context.eager_mode():
142 with context.eager_mode():
152 with context.eager_mode():
193 with context.eager_mode():
217 with context.eager_mode():
[all …]
|
| D | resource_variable_ops_test.py | 90 with context.eager_mode(), context.device("gpu:0"):
95 with context.eager_mode():
102 with context.eager_mode():
107 with context.eager_mode():
118 with context.eager_mode():
130 with context.eager_mode():
135 with context.eager_mode():
208 with context.eager_mode():
220 with context.eager_mode():
227 with context.eager_mode():
[all …]
|
| D | garbage_collection_test.py | 36 with context.eager_mode():
41 with context.eager_mode():
|
| D | reduce_benchmark_test.py | 55 with context.eager_mode():
64 with context.eager_mode(), ops.device("/cpu:0"):
|
| D | template_test.py | 120 with context.eager_mode():
203 with context.eager_mode():
303 with context.eager_mode():
452 with context.eager_mode():
|
| D | logging_ops_test.py | 324 with context.eager_mode():
332 with context.eager_mode():
|
| D | variable_scope_test.py | 185 with context.eager_mode():
195 with context.eager_mode():
233 with context.eager_mode():
276 with context.eager_mode():
283 with context.eager_mode():
1468 with context.eager_mode():
1532 with context.eager_mode():
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | xplane_to_tf_functions_test.cc | 167 const auto& eager_mode = eager_metrics.at(EAGER_MODE); in TEST() local
168 EXPECT_EQ(eager_mode.count(), 1); in TEST()
169 EXPECT_EQ(eager_mode.self_time_ps(), 200); in TEST()
|
| /external/tensorflow/tensorflow/python/data/benchmarks/ |
| D | meta_benchmark.py | 57 with context.eager_mode():
80 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/autograph/utils/ |
| D | tensor_list_test.py | 57 with context.eager_mode():
71 with context.eager_mode():
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.__internal__.eager_context.pbtxt | 4 name: "eager_mode"
|
| /external/tensorflow/tensorflow/python/eager/memory_tests/ |
| D | memory_test_util.py | 51 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | keras_parameterized.py | 428 with context.eager_mode():
434 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | arithmetic_optimizer_test.py | 38 with context.eager_mode():
|
| D | constant_folding_test.py | 90 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | combinations.py | 46 return [context.eager_mode()]
|
| D | ops_test.py | 198 with context.eager_mode():
491 with context.eager_mode():
796 with context.eager_mode():
1627 with context.eager_mode():
1797 with context.eager_mode():
2196 with context.eager_mode():
2733 with context.eager_mode():
2746 with context.eager_mode():
2748 self.assertTrue(context.eager_mode())
2749 self.assertTrue(context.eager_mode())
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | multi_worker_test_base.py | 533 def _run_client(self, client_fn, task_type, task_id, num_gpus, eager_mode, argument
540 if eager_mode:
541 with context.eager_mode():
688 with context.eager_mode():
|
| D | device_util_test.py | 60 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adam_test.py | 255 with context.eager_mode():
347 with context.eager_mode():
372 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_benchmark.py | 68 with context.eager_mode(), _get_strategy(num_gpus).scope() as strategy:
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 678 bool eager_mode) { in GetMklShape() argument
679 if (!eager_mode) { in GetMklShape()
724 bool eager_mode = false) {
730 GetMklShape(context, input_idx, &input_mkl_shape, eager_mode);
731 if (input_mkl_shape.IsMklTensor() && !eager_mode) {
760 bool eager_mode = false) {
764 if (!eager_mode) {
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | memory_cleanup_test.py | 54 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | tracking.py | 196 context.eager_mode if context.executing_eagerly()
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | saver_test.py | 148 with context.eager_mode():
|
123