| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | summary_ops_test.py | 58 with context.eager_mode():
71 with context.eager_mode():
90 with context.eager_mode():
111 with context.eager_mode():
120 with context.eager_mode():
131 with context.eager_mode():
146 with context.eager_mode():
156 with context.eager_mode():
189 with context.eager_mode():
213 with context.eager_mode():
[all …]
|
| D | resource_variable_ops_test.py | 89 with context.eager_mode(), context.device("gpu:0"):
94 with context.eager_mode():
101 with context.eager_mode():
106 with context.eager_mode():
117 with context.eager_mode():
129 with context.eager_mode():
134 with context.eager_mode():
199 with context.eager_mode():
211 with context.eager_mode():
374 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 | 121 with context.eager_mode():
204 with context.eager_mode():
305 with context.eager_mode():
477 with context.eager_mode():
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_grad_input_ops.cc | 300 template <typename Device, class T, bool is_depthwise, bool eager_mode>
324 GetMklShape(context, kInputIdx, &src_mkl_shape, eager_mode); in Compute()
325 GetMklShape(context, kFilterIdx, &filter_mkl_shape, eager_mode); in Compute()
326 GetMklShape(context, kOutbpropIdx, &diff_dst_mkl_shape, eager_mode); in Compute()
338 GetTfShape(context, kOutbpropIdx, eager_mode); in Compute()
352 eager_mode); in Compute()
437 if (eager_mode) { in Compute()
442 diff_src_mkl_shape, eager_mode); in Compute()
469 if (!eager_mode) { in Compute()
535 return GetTfShape(context, kInputIndex_Filter, eager_mode); in MakeFilterTfShape()
|
| D | mkl_conv_grad_filter_ops.cc | 361 bool eager_mode>
386 GetMklShape(context, kInputIdx, &src_mkl_shape, eager_mode); in Compute()
387 GetMklShape(context, kFilterIdx, &filter_mkl_shape, eager_mode); in Compute()
388 GetMklShape(context, kOutbpropIdx, &diff_dst_mkl_shape, eager_mode); in Compute()
400 GetTfShape(context, kOutbpropIdx, eager_mode); in Compute()
414 eager_mode); in Compute()
500 eager_mode); in Compute()
626 return GetTfShape(context, input_idx, eager_mode); in MakeInputTfShape()
|
| D | mkl_conv_ops.cc | 445 bool eager_mode>
517 GetMklShape(context, kInputIndex_Src, &src_mkl_shape, eager_mode); in Compute()
518 GetMklShape(context, kInputIndex_Filter, &filter_mkl_shape, eager_mode); in Compute()
549 auto src_tf_shape = GetTfShape(context, kInputIndex_Src, eager_mode); in Compute()
551 GetTfShape(context, kInputIndex_Filter, eager_mode); in Compute()
569 !eager_mode; in Compute()
574 src_tf_shape, dst_mkl_shape, eager_mode); in Compute()
758 if (!eager_mode) { in Compute()
924 if (eager_mode) { in AllocateOutputTensor()
942 eager_mode); in AllocateOutputTensor()
[all …]
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/ |
| D | loss_scale_benchmark.py | 74 ctx_mgr = context.eager_mode()
77 ctx_mgr = context.eager_mode()
|
| /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 | 58 with context.eager_mode():
72 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | add_loss_correctness_test.py | 107 with context.eager_mode():
127 with context.eager_mode():
167 with context.eager_mode():
442 with context.eager_mode():
453 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/eager/memory_tests/ |
| D | memory_test_util.py | 51 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | multi_worker_test_base.py | 327 def _run_client(self, client_fn, task_type, task_id, num_gpus, eager_mode, argument
334 if eager_mode:
335 with context.eager_mode():
482 with context.eager_mode():
|
| D | device_util_test.py | 49 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | metrics_functional_test.py | 73 with context.eager_mode():
81 with context.eager_mode():
|
| D | keras_parameterized.py | 409 with context.eager_mode():
416 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | arithmetic_optimizer_test.py | 38 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | combinations.py | 44 return [context.eager_mode()]
|
| D | ops_test.py | 199 with context.eager_mode():
331 with context.eager_mode():
636 with context.eager_mode():
1442 with context.eager_mode():
1612 with context.eager_mode():
2011 with context.eager_mode():
2548 with context.eager_mode():
2561 with context.eager_mode():
2563 self.assertTrue(context.eager_mode())
2564 self.assertTrue(context.eager_mode())
[all …]
|
| /external/tensorflow/tensorflow/python/keras/premade/ |
| D | linear_test.py | 96 with context.eager_mode():
136 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_utils_test.py | 57 with context.eager_mode():
77 with context.eager_mode():
94 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adam_test.py | 251 with context.eager_mode():
339 with context.eager_mode():
362 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | dense_features_test.py | 58 with context.eager_mode():
103 with context.eager_mode():
155 with context.eager_mode():
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | tensorflow_op_layer_test.py | 351 with context.eager_mode():
360 with context.eager_mode():
|