Searched refs:eager_context (Results 1 – 11 of 11) sorted by relevance
21 TraceCollector::TraceCollector(EagerContext* const eager_context) in TraceCollector() argument22 : context_(eager_context) {} in TraceCollector()34 EagerContext* const eager_context) { in Create() argument35 return absl::WrapUnique(new EagerProfiler(eager_context)); in Create()56 EagerProfiler::EagerProfiler(EagerContext* const eager_context) in EagerProfiler() argument57 : context_(eager_context), collector_(eager_context) {} in EagerProfiler()
27 TraceCollector(EagerContext* const eager_context);41 EagerContext* const eager_context);50 EagerProfiler(EagerContext* const eager_context);
59 TFE_Context* eager_context) { in TFE_ProfilerContextSetEagerContext() argument60 profiler_context->profiler_context.eager_context = &eager_context->context; in TFE_ProfilerContextSetEagerContext()
53 TFE_ProfilerContext* profiler_context, TFE_Context* eager_context);
28 Env* env = profiler_context->eager_context != nullptr in StartProfilerServer()29 ? profiler_context->eager_context->TFEnv() in StartProfilerServer()
47 Env* env = profiler_context_.eager_context != nullptr in Profile()48 ? profiler_context_.eager_context->TFEnv() in Profile()
235 tensorflow::Status BuildEagerOp(tensorflow::EagerContext* eager_context) { in BuildEagerOp() argument250 op_.reset(new tensorflow::EagerOperation(eager_context, name_.c_str(), in BuildEagerOp()362 tensorflow::EagerContext* eager_context; member377 op_data->eager_context = in Init()413 status = node_data.BuildEagerOp(op_data->eager_context); in Init()454 context, op_data->eager_context != nullptr, in Prepare()
563 eager_context = self._thread_local_data564 old_device_name = eager_context.device_name565 old_device_spec = eager_context.device_spec593 eager_context.device_name = new_device_name594 eager_context.device_spec = new_device_spec597 eager_context.device_name = old_device_name598 eager_context.device_spec = old_device_spec
166 if (context->eager_context != nullptr) { in ProfilerSession()168 context->eager_context)); in ProfilerSession()
26 EagerContext* eager_context = nullptr; member
31 from tensorflow.python.eager import context as eager_context unknown1394 ctx = eager_context.context()1405 ctx = eager_context.context()1658 if eager_context.executing_eagerly():1667 if eager_context.executing_eagerly():