| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 46 EagerContext::EagerContext(const SessionOptions& opts, in EagerContext() function in tensorflow::EagerContext
51 : EagerContext(opts, default_policy, async, device_mgr.release(), in EagerContext()
54 EagerContext::EagerContext(const SessionOptions& opts, in EagerContext() function in tensorflow::EagerContext
93 void EagerContext::InitDeviceMapAndAsync() { in InitDeviceMapAndAsync()
118 bool EagerContext::Async() const { in Async()
124 Status EagerContext::SetAsyncForThread(bool async) { in SetAsyncForThread()
141 Status EagerContext::ClearCaches() { in ClearCaches()
152 void EagerContext::SetThreadLocalDevicePlacementPolicy( in SetThreadLocalDevicePlacementPolicy()
158 ContextDevicePlacementPolicy EagerContext::GetDevicePlacementPolicy() { in GetDevicePlacementPolicy()
168 void EagerContext::CloseRemoteContexts() { in CloseRemoteContexts()
[all …]
|
| D | tensor_handle.h | 61 EagerContext* ctx);
63 Device* resource_device, DataType dtype, EagerContext* ctx);
68 Device* op_device, Device* resource_device, EagerContext* ctx);
105 Status CopyToDevice(EagerContext* ctx, tensorflow::Device* dstd,
109 EagerContext* Context() { in Context()
186 EagerContext* ctx_ GUARDED_BY(ctx_mutex_);
197 Device* GetResourceDevice(const Tensor& t, EagerContext* ctx);
|
| D | execute.cc | 57 void MaybeInitializeStepStats(StepStats* step_stats, EagerContext* ctx) { in MaybeInitializeStepStats()
67 int StepStatsDeviceIndex(StepStats* step_stats, EagerContext* ctx, in StepStatsDeviceIndex()
96 EagerContext* ctx = op->EagerContext(); in MaybeCopyInputToExpectedDevice()
180 Status ValidateInputTypeAndPlacement(EagerContext* ctx, in ValidateInputTypeAndPlacement()
205 Status SelectDevice(const NodeDef& ndef, EagerContext* ctx, Device** device) { in SelectDevice()
230 op->EagerContext()->FuncLibDef()->Find(op->Name()); in GetOutputDTypes()
245 bool IsLocal(EagerContext* ctx, tensorflow::Device* d) { in IsLocal()
251 bool OnSameTask(EagerContext* ctx, Device* first, Device* second) { in OnSameTask()
260 Status CPUDeviceOnTask(EagerContext* ctx, tensorflow::Device* device, in CPUDeviceOnTask()
275 Status FindDeviceFromName(const EagerContext* ctx, const char* device_name, in FindDeviceFromName()
[all …]
|
| D | eager_operation.h | 25 EagerOperation(tensorflow::EagerContext* ctx, const char* op, in EagerOperation()
42 tensorflow::EagerContext* EagerContext() { return ctx_; } in EagerContext() function
70 tensorflow::EagerContext* ctx_; // Must outlive the EagerOperation.
|
| D | execute.h | 46 Status EagerKernelExecute(EagerContext* ctx,
54 Status EagerCopyToDevice(TensorHandle* h, EagerContext* ctx,
|
| D | tensor_handle.cc | 49 EagerContext* ctx) in TensorHandle()
64 EagerContext* ctx) in TensorHandle()
85 EagerContext* ctx) in TensorHandle()
251 Status TensorHandle::CopyToDevice(EagerContext* ctx, tensorflow::Device* dstd, in CopyToDevice()
318 Device* GetResourceDevice(const Tensor& t, EagerContext* ctx) { in GetResourceDevice()
|
| D | copy_to_device_node.h | 28 CopyToDeviceNode(TensorHandle* src, Device* dstd, EagerContext* ctx) in CopyToDeviceNode()
61 EagerContext* ctx_;
|
| D | context.h | 76 class EagerContext {
79 EagerContext(const SessionOptions& opts,
84 EagerContext(const SessionOptions& opts,
89 ~EagerContext();
|
| D | execute_node.h | 34 ExecuteNode(uint64 id, EagerContext* ctx, in ExecuteNode()
80 tensorflow::EagerContext* ctx_;
|
| /external/tensorflow/tensorflow/core/profiler/internal/runtime/ |
| D | eager_profiler.h | 27 TraceCollector(EagerContext* const eager_context);
35 EagerContext* const context_;
41 EagerContext* const eager_context);
50 EagerProfiler(EagerContext* const eager_context);
56 EagerContext* const context_;
|
| D | eager_profiler.cc | 21 TraceCollector::TraceCollector(EagerContext* const eager_context) in TraceCollector()
34 EagerContext* const eager_context) { in Create()
56 EagerProfiler::EagerProfiler(EagerContext* const eager_context) in EagerProfiler()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_data.h | 42 tensorflow::EagerContext* GetEagerContext() { return eager_context_.get(); } in GetEagerContext()
51 std::unique_ptr<tensorflow::EagerContext> eager_context_;
|
| D | delegate_data.cc | 43 eager_context_.reset(new tensorflow::EagerContext( in Prepare()
|
| D | kernel.cc | 235 tensorflow::Status BuildEagerOp(tensorflow::EagerContext* eager_context) { in BuildEagerOp()
362 tensorflow::EagerContext* eager_context;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.h | 107 explicit ServerContext(std::unique_ptr<tensorflow::EagerContext> ctx, in ServerContext()
120 tensorflow::EagerContext* Context() const { return ctx_.get(); } in Context()
182 std::unique_ptr<tensorflow::EagerContext> ctx_;
|
| D | eager_service_impl.cc | 46 Status GetNumRetvals(tensorflow::EagerContext* context, const string& op_name, in GetNumRetvals()
120 std::unique_ptr<tensorflow::EagerContext> ctx(new tensorflow::EagerContext( in CreateContext()
|
| /external/tensorflow/tensorflow/core/profiler/lib/ |
| D | profiler_session.h | 26 EagerContext* eager_context = nullptr;
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_internal.h | 75 tensorflow::EagerContext context;
|
| D | c_api.cc | 601 ? op->operation.EagerContext()->HostCPU() in TFE_OpGetDevice()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 8784 n->thread.reset(op->operation.EagerContext()->TFEnv()->StartThread( in TFE_ExecuteOpInNewThread()
|