Home
last modified time | relevance | path

Searched refs:EagerContext (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.h65 Device* d, Device* op_device, EagerContext* ctx);
69 Device* op_device, EagerContext* ctx);
72 DataType dtype, EagerContext* ctx);
76 Device* d, Device* resource_device, EagerContext* ctx);
78 DataType dtype, Device* device, EagerContext* ctx);
86 EagerContext* ctx, TensorHandle** h);
88 Device* op_device, EagerContext* ctx,
92 EagerContext* ctx, TensorHandle** h);
98 Device* resource_device, EagerContext* ctx,
102 Device* resource_device, EagerContext* ctx,
[all …]
Dcontext.cc72 EagerContext::EagerContext( in EagerContext() function in tensorflow::EagerContext
126 void EagerContext::ResetPFLR(const DeviceMgr* device_mgr, Env* env, in ResetPFLR()
144 void EagerContext::InitPrioritizedDeviceTypeList() { in InitPrioritizedDeviceTypeList()
171 Status EagerContext::SelectDevice(DeviceNameUtils::ParsedName preferred, in SelectDevice()
232 void EagerContext::ResetClusterFLR( in ResetClusterFLR()
237 EagerExecutor& EagerContext::Executor() { in Executor()
243 void EagerContext::SetExecutorForThread(EagerExecutor* executor) { in SetExecutorForThread()
252 void EagerContext::ClearCachesAndThreadExecutors() { in ClearCachesAndThreadExecutors()
264 void EagerContext::ClearCachesAndDefaultExecutor() { in ClearCachesAndDefaultExecutor()
276 void EagerContext::SetThreadLocalDevicePlacementPolicy( in SetThreadLocalDevicePlacementPolicy()
[all …]
Dtensor_handle.cc103 EagerContext* ctx, TensorHandle** h) { in CreateLocalHandle()
108 Device* op_device, EagerContext* ctx, in CreateLocalHandle()
124 EagerContext* ctx) in TensorHandle()
143 Device* op_device, EagerContext* ctx) in TensorHandle()
164 DataType dtype, EagerContext* ctx, in CreateEmptyLocalHandle()
175 EagerContext* ctx) in TensorHandle()
196 Device* resource_device, EagerContext* ctx, TensorHandle** h) { in CreateRemoteHandle()
207 EagerContext* ctx, TensorHandle** h) { in CreateRemoteHandle()
217 EagerContext* ctx) in TensorHandle()
235 Device* d, EagerContext* ctx, TensorHandle** h) { in CreateUnshapedRemoteHandle()
[all …]
Deager_op_rewrite_registry_test.cc33 new tensorflow::EagerOperation(&orig_op->EagerContext()); in Run()
48 tensorflow::EagerContext* ctx = new tensorflow::EagerContext( in TEST()
Dexecute.cc93 Status CopyInputToExpectedDevice(EagerContext* ctx, EagerOperation* op, in CopyInputToExpectedDevice()
166 EagerContext* ctx, EagerOperation* op, in ValidateInputTypeAndPlacement()
206 op->EagerContext().FuncLibDef()->Find(op->Name()); in GetOutputDTypes()
230 Status GetDeviceForInput(const EagerContext& ctx, TensorHandle* tensor_handle, in GetDeviceForInput()
279 Status MustCompileWithXLA(const EagerOperation* op, const EagerContext& ctx, in MustCompileWithXLA()
353 EagerContext& ctx = op->EagerContext(); in EagerLocalExecute()
596 EagerContext& ctx = op->EagerContext(); in PrepareRemoteOp()
630 EagerContext& ctx = op->EagerContext(); in EagerRemoteExecute()
774 op->MutableAttrs()->BuildNodeDef(), op->EagerContext().FuncLibDef(), in EagerRemoteExecute()
827 EagerContext& ctx = op->EagerContext(); in MaybeUpdateOpDevice()
[all …]
Deager_operation.h31 explicit EagerOperation(tensorflow::EagerContext* ctx) : ctx_(*ctx) {} in EagerOperation()
56 tensorflow::EagerContext& EagerContext() { return ctx_; } in EagerContext() function
135 tensorflow::EagerContext& ctx_;
Dexecute_node.h51 Status Init(EagerContext* ctx,
75 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& inputs, in ExecuteNode()
146 EagerContext* ctx_;
Dexecute.h51 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs,
62 Status EagerCopyToDevice(TensorHandle* h, EagerContext* ctx,
Dmkl_eager_op_rewrite_test.cc39 std::unique_ptr<tensorflow::EagerContext> eager_ctx = in CreateOp()
40 std::unique_ptr<tensorflow::EagerContext>(new tensorflow::EagerContext( in CreateOp()
Dcopy_to_device_node.h30 const EagerContext& ctx) in CopyToDeviceNode()
65 const EagerContext& ctx_;
Dcontext_test.cc51 EagerContext* context() { return context_; } in context()
57 context_ = new EagerContext( in InitContext()
81 EagerContext* context_;
Dexecute_node.cc21 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs) { in Init()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.h88 EagerContext* context);
112 static ServerContext* CreateMasterContext(tensorflow::EagerContext* ctx, in CreateMasterContext()
117 explicit ServerContext(tensorflow::EagerContext* ctx,
137 tensorflow::EagerContext* Context() const { return ctx_; } in Context()
153 tensorflow::EagerContext* ctx_;
203 Status ExecuteOp(const Operation& operation, EagerContext* eager_context,
207 EagerContext* eager_context);
209 EagerContext* eager_context);
Dremote_tensor_handle_data.h29 EagerContext* ctx);
53 EagerContext* const ctx_;
62 EagerContext* ctx);
80 EagerContext* ctx() const { return ctx_; } in ctx()
98 EagerContext* const ctx_;
Dcluster_function_library_runtime.h37 EagerClusterFunctionLibraryRuntime(const uint64 context_id, EagerContext* ctx, in EagerClusterFunctionLibraryRuntime()
68 EagerContext* ctx_;
90 const uint64 context_id, EagerContext* ctx, WorkerSession* worker_session);
Deager_service_impl.cc55 Status GetNumRetvals(tensorflow::EagerContext* context, const string& op_name, in GetNumRetvals()
163 tensorflow::EagerContext* ctx = new tensorflow::EagerContext( in CreateContext()
231 tensorflow::EagerContext* ctx = server_context->Context(); in UpdateContext()
312 const tensorflow::uint64 context_id, EagerContext* context) { in CreateMasterContext()
341 EagerContext* eager_context, in ExecuteOp()
469 tensorflow::EagerContext* ctx = context->Context(); in KeepAlive()
506 const RegisterFunctionOp& register_function, EagerContext* eager_context) { in RegisterFunction()
521 EagerContext* eager_context) { in SendTensor()
Dremote_mgr_test.cc32 TestRemoteMgr(bool is_master, EagerContext* ctx) in TestRemoteMgr()
55 ctx_ = new tensorflow::EagerContext( in RemoteMgrTest()
68 EagerContext* ctx_;
Dremote_tensor_handle_data.cc27 void DestroyRemoteTensorHandle(EagerContext* ctx, const string& remote_task, in DestroyRemoteTensorHandle()
90 EagerContext* ctx) in RemoteTensorHandleData()
152 EagerContext* ctx) in UnshapedRemoteTensorHandleData()
Dremote_copy_node.h63 RemoteCopyNode(EagerContext* ctx, EagerExecutor* executor, TensorHandle* src,
161 EagerContext* const ctx_;
Dremote_mgr.h33 RemoteMgr(bool is_master, EagerContext* ctx) in RemoteMgr()
119 EagerContext* parent_; // not owned.
Dremote_copy_node.cc43 EagerContext& ctx = op->EagerContext(); in CreateUncachedKernelAndDeviceOp()
73 RemoteCopyNode::RemoteCopyNode(EagerContext* ctx, EagerExecutor* executor, in RemoteCopyNode()
Deager_service_impl_test.cc52 Status GetEagerContext(const uint64 context_id, EagerContext** ctx) { in GetEagerContext()
482 EagerContext* ctx = nullptr; in Init()
606 EagerContext* ctx = nullptr; in TEST_F()
712 tensorflow::EagerContext* ctx = new tensorflow::EagerContext( in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_data.h42 tensorflow::EagerContext* GetEagerContext() { return eager_context_; } in GetEagerContext()
51 tensorflow::EagerContext* eager_context_ = nullptr;
Ddelegate_data.cc46 eager_context_ = new tensorflow::EagerContext( in Prepare()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc474 tensorflow::EagerContext* context = ctx->context; in UpdateTFE_ContextWithServerDef()
495 context_id = tensorflow::EagerContext::NewContextId(); in UpdateTFE_ContextWithServerDef()
710 return new TFE_Context{new tensorflow::EagerContext( in TFE_NewContext()
728 return new TFE_Context{new tensorflow::EagerContext( in TFE_NewContextFromSession()
808 tensorflow::EagerContext::kInvalidContextId) { in TFE_ContextUpdateServerDef()
831 tensorflow::EagerContext* context = ctx->context; in TFE_ContextCheckAlive()
1095 tensorflow::EagerContext* ctx = handle_->Context(); in Resolve()
1141 tensorflow::EagerContext* context = ctx->context; in TFE_NewTensorHandleFromDeviceMemory()
1226 ? op->operation.EagerContext().HostCPU() in TFE_OpGetDevice()
1508 tensorflow::EagerContext* context = ctx->context; in TFE_TensorHandleCopyToDevice()
[all …]

12