/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | tensor_handle.h | 65 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 …]
|
D | context.cc | 72 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 …]
|
D | tensor_handle.cc | 103 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 …]
|
D | eager_op_rewrite_registry_test.cc | 33 new tensorflow::EagerOperation(&orig_op->EagerContext()); in Run() 48 tensorflow::EagerContext* ctx = new tensorflow::EagerContext( in TEST()
|
D | execute.cc | 93 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 …]
|
D | eager_operation.h | 31 explicit EagerOperation(tensorflow::EagerContext* ctx) : ctx_(*ctx) {} in EagerOperation() 56 tensorflow::EagerContext& EagerContext() { return ctx_; } in EagerContext() function 135 tensorflow::EagerContext& ctx_;
|
D | execute_node.h | 51 Status Init(EagerContext* ctx, 75 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& inputs, in ExecuteNode() 146 EagerContext* ctx_;
|
D | execute.h | 51 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs, 62 Status EagerCopyToDevice(TensorHandle* h, EagerContext* ctx,
|
D | mkl_eager_op_rewrite_test.cc | 39 std::unique_ptr<tensorflow::EagerContext> eager_ctx = in CreateOp() 40 std::unique_ptr<tensorflow::EagerContext>(new tensorflow::EagerContext( in CreateOp()
|
D | copy_to_device_node.h | 30 const EagerContext& ctx) in CopyToDeviceNode() 65 const EagerContext& ctx_;
|
D | context_test.cc | 51 EagerContext* context() { return context_; } in context() 57 context_ = new EagerContext( in InitContext() 81 EagerContext* context_;
|
D | execute_node.cc | 21 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs) { in Init()
|
/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
D | eager_service_impl.h | 88 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);
|
D | remote_tensor_handle_data.h | 29 EagerContext* ctx); 53 EagerContext* const ctx_; 62 EagerContext* ctx); 80 EagerContext* ctx() const { return ctx_; } in ctx() 98 EagerContext* const ctx_;
|
D | cluster_function_library_runtime.h | 37 EagerClusterFunctionLibraryRuntime(const uint64 context_id, EagerContext* ctx, in EagerClusterFunctionLibraryRuntime() 68 EagerContext* ctx_; 90 const uint64 context_id, EagerContext* ctx, WorkerSession* worker_session);
|
D | eager_service_impl.cc | 55 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()
|
D | remote_mgr_test.cc | 32 TestRemoteMgr(bool is_master, EagerContext* ctx) in TestRemoteMgr() 55 ctx_ = new tensorflow::EagerContext( in RemoteMgrTest() 68 EagerContext* ctx_;
|
D | remote_tensor_handle_data.cc | 27 void DestroyRemoteTensorHandle(EagerContext* ctx, const string& remote_task, in DestroyRemoteTensorHandle() 90 EagerContext* ctx) in RemoteTensorHandleData() 152 EagerContext* ctx) in UnshapedRemoteTensorHandleData()
|
D | remote_copy_node.h | 63 RemoteCopyNode(EagerContext* ctx, EagerExecutor* executor, TensorHandle* src, 161 EagerContext* const ctx_;
|
D | remote_mgr.h | 33 RemoteMgr(bool is_master, EagerContext* ctx) in RemoteMgr() 119 EagerContext* parent_; // not owned.
|
D | remote_copy_node.cc | 43 EagerContext& ctx = op->EagerContext(); in CreateUncachedKernelAndDeviceOp() 73 RemoteCopyNode::RemoteCopyNode(EagerContext* ctx, EagerExecutor* executor, in RemoteCopyNode()
|
D | eager_service_impl_test.cc | 52 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/ |
D | delegate_data.h | 42 tensorflow::EagerContext* GetEagerContext() { return eager_context_; } in GetEagerContext() 51 tensorflow::EagerContext* eager_context_ = nullptr;
|
D | delegate_data.cc | 46 eager_context_ = new tensorflow::EagerContext( in Prepare()
|
/external/tensorflow/tensorflow/c/eager/ |
D | c_api.cc | 474 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 …]
|