Home
last modified time | relevance | path

Searched refs:is_eager (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/utils/
Dop_utils.cc44 absl::string_view category, bool is_eager, in EnterOp() argument
51 op_metrics->set_is_eager(op_metrics->is_eager() || is_eager); in EnterOp()
69 absl::string_view provenance, bool is_eager, uint64 occurrences, in EnterOp() argument
81 op_metrics->set_is_eager(op_metrics->is_eager() || is_eager); in EnterOp()
Dop_utils.h42 bool is_eager, uint64 time_ps, uint64 children_time_ps);
72 bool is_eager, uint64 occurrences, uint64 time_ps,
Dop_metrics_db_utils.cc50 tf_op_metrics->set_is_eager(device_op_metrics.is_eager()); in UpdateTfOpMetricsWithDeviceOpMetrics()
Dgroup_events.h102 void SetIsEager(bool is_eager);
Dgroup_events.cc485 void EventNode::SetIsEager(bool is_eager) { in SetIsEager() argument
486 FindOrAddStatByType(StatType::kIsEager)->set_int64_value(is_eager ? 1 : 0); in SetIsEager()
/external/tensorflow/tensorflow/core/profiler/convert/
Dxplane_to_op_metrics_db.cc62 bool is_eager; member
98 activity.tf_op.name, activity.tf_op.type, activity.is_eager, in ProcessOneTfActivity()
150 bool is_eager = false; in CollectTfActivities() local
153 is_eager = stat->IntValue(); in CollectTfActivities()
157 {span.begin_ps(), tf_op_id, kTfOpBegin, *tf_op, is_eager}); in CollectTfActivities()
159 {span.end_ps(), tf_op_id, kTfOpEnd, *tf_op, is_eager}); in CollectTfActivities()
229 bool is_eager; in ConvertDeviceTraceXPlaneToOpMetricsDb() local
235 is_eager = stat.IntValue(); in ConvertDeviceTraceXPlaneToOpMetricsDb()
246 tf_op.type, tf_op_full_name, is_eager, in ConvertDeviceTraceXPlaneToOpMetricsDb()
Dop_metrics_db_combiner.cc66 dst->set_is_eager(dst->is_eager() || src.is_eager()); in CombineOpMetrics()
Dop_stats_to_tf_stats.cc41 record.set_is_eager(metrics.is_eager()); in ConvertOpMetricsToTfStatsRecord()
Dop_stats_to_overview_page.cc222 if (metrics.is_eager()) eager_host_op_time_ps += metrics.self_time_ps(); in ComputeAnalysisResult()
232 if (metrics.is_eager()) eager_device_op_time_ps += metrics.self_time_ps(); in ComputeAnalysisResult()
/external/tensorflow/tensorflow/python/framework/
Dpython_tensor_converter_wrapper.cc72 bool is_eager = PyObject_IsTrue(py_is_eager.get()); in MakePythonTensorConverter() local
77 if (is_eager) { in MakePythonTensorConverter()
Dop_callbacks_test.py84 is_eager = not graph
85 if is_eager:
Dtest_util.py3482 is_eager = context.executing_eagerly()
3483 if not is_eager and run_eagerly:
Dops.py5901 c._thread_local_data.is_eager = False # pylint: disable=protected-access
5970 context._context._thread_local_data.is_eager = True
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe.h413 bool is_eager = false; member
430 PyObject* is_eager,
Dcontext.py404 is_eager=lambda: default_execution_mode == EAGER_MODE,
811 old_is_eager = ctx.is_eager
812 ctx.is_eager = mode == EAGER_MODE
821 ctx.is_eager = old_is_eager
827 return self._thread_local_data.is_eager
Dpywrap_tfe_src.cc4271 tensorflow::Safe_PyObjectPtr is_eager; member
4297 PyObject* is_eager, argument
4313 Py_INCREF(is_eager);
4314 defaults.is_eager.reset(is_eager);
4346 Safe_PyObjectPtr is_eager(PyObject_CallFunctionObjArgs( local
4347 defaults->second.is_eager.get(), nullptr));
4348 if (!is_eager) return nullptr;
4349 thread_local_data->is_eager = PyObject_IsTrue(is_eager.get());
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_3d_test.py287 is_eager = ctx is not None and ctx.executing_eagerly()
289 (test_util.IsMklEnabled() and is_eager is False):
315 is_eager = ctx is not None and ctx.executing_eagerly()
317 (test_util.IsMklEnabled() and is_eager is False):
/external/tensorflow/tensorflow/core/profiler/protobuf/
Dop_metrics.proto42 bool is_eager = 18; field
Dtf_stats.proto75 bool is_eager = 18; field
/external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/
Drnn_cell_impl.py332 is_eager = context.executing_eagerly()
333 if is_eager and _hasattr(self, "_last_zero_state"):
341 if is_eager:
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc379 py::handle is_eager, in EagerContextThreadLocalDataWrapper() argument
383 py_eager_context.ptr(), is_eager.ptr(), device_spec.ptr()); in EagerContextThreadLocalDataWrapper()
390 bool get_is_eager() const { return GetData()->is_eager; } in get_is_eager()
391 void set_is_eager(bool v) { GetData()->is_eager = v; } in set_is_eager()