| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 72 EagerContext* global_c_eager_context = nullptr;
76 void SetCEagerContext(EagerContext* ctx) { global_c_eager_context = ctx; } in SetCEagerContext()
78 EagerContext* GetCEagerContext() { return global_c_eager_context; } in GetCEagerContext()
96 const int64_t EagerContext::kGlobalRendezvousId = -1;
100 IntraProcessRendezvous* EagerContext::LocalRendezvousTable::FindOrCreate( in FindOrCreate()
108 if (step_id == EagerContext::kGlobalRendezvousId) { in FindOrCreate()
116 IntraProcessRendezvous* EagerContext::LocalRendezvousTable::Find( in Find()
125 void EagerContext::LocalRendezvousTable::Remove(int64_t step_id) { in Remove()
133 void EagerContext::LocalRendezvousTable::CleanUpAll() { in CleanUpAll()
145 EagerContext::LocalRendezvousTable::~LocalRendezvousTable() { CleanUpAll(); } in ~LocalRendezvousTable()
[all …]
|
| D | tensor_handle.h | 51 class EagerContext; variable
59 Device* resource_device, EagerContext* ctx);
62 EagerContext* ctx);
64 tensorflow::DataType dtype, EagerContext* ctx);
68 tensorflow::DataType dtype, Device* device, EagerContext* ctx,
71 Device* device, const bool is_ready, EagerContext* ctx);
78 Device* op_device, EagerContext* ctx);
82 EagerContext* ctx);
86 EagerContext* ctx);
98 const string& device_name, EagerContext* ctx,
[all …]
|
| D | execute_node.h | 54 Status Init(EagerContext* ctx,
75 bool IsRemote(EagerContext* ctx, Device* input_device, TensorHandle* handle);
79 Status InitPackedHandle(const int index, EagerContext* ctx,
94 ExecuteNode(EagerContext* ctx, in ExecuteNode()
138 EagerContext* ctx_;
150 AsyncExecuteNode(EagerContext* ctx, in AsyncExecuteNode()
232 EagerContext* ctx_;
|
| D | core.cc | 106 ImmediateExecutionTensorHandle* EagerContext::CopyTensorHandleToDevice( in CopyTensorHandleToDevice()
132 ImmediateExecutionTensorHandle* EagerContext::CreateLocalHandle( in CreateLocalHandle()
139 ImmediateExecutionTensorHandle* EagerContext::CreateLocalHandleFromTFTensor( in CreateLocalHandleFromTFTensor()
152 ImmediateExecutionTensorHandle* EagerContext::TFTensorHandleFromInterface( in TFTensorHandleFromInterface()
160 ImmediateExecutionOperation* EagerContext::CreateOperation() { in CreateOperation()
164 Status EagerContext::RegisterFunction(AbstractFunction* f) { in RegisterFunction()
|
| D | execute.cc | 122 Status CopyInputToExpectedDevice(EagerContext* ctx, EagerOperation* op, in CopyInputToExpectedDevice()
209 EagerContext* ctx, EagerOperation* op, in ValidateInputTypeAndPlacement()
272 op->EagerContext().FuncLibDef()->Find(op->Name()); in GetOutputDTypes()
320 Status GetDeviceForInput(const EagerOperation& op, const EagerContext& ctx, in GetDeviceForInput()
388 Status GetFuncAttr(const EagerOperation* op, const EagerContext& ctx, in GetFuncAttr()
412 Status MustCompileWithXLA(const EagerOperation* op, const EagerContext& ctx, in MustCompileWithXLA()
818 if (!op->EagerContext().GetFunctionDef(fname)) { in WrapInCallOp()
855 TF_RETURN_IF_ERROR(op->EagerContext().AddFunctionDef(std::move(fdef))); in WrapInCallOp()
858 auto& ctx = op->EagerContext(); in WrapInCallOp()
910 EagerContext& ctx = op.EagerContext(); in GetKernelCacheKey()
[all …]
|
| D | tensor_handle_test.cc | 39 auto ctx = new EagerContext( in TEST()
105 context_ = new EagerContext( in PackedTensorHandleTest()
118 EagerContext* context() { return context_; } in context()
136 EagerContext* context_;
255 auto ctx = new EagerContext( in TEST()
287 auto ctx = new EagerContext( in TEST()
341 context_ = new EagerContext( in RemoteTensorHandleTest()
354 EagerContext* context() { return context_; } in context()
367 EagerContext* context_;
380 EagerContext* context = new EagerContext( in TEST_F()
[all …]
|
| D | context_distributed_manager.h | 35 class EagerContext; variable
41 explicit EagerContextDistributedManager(EagerContext* context) in EagerContextDistributedManager()
63 EagerContext* context_;
|
| D | custom_device_test.cc | 100 core::RefCountPtr<EagerContext> ctx(new EagerContext( in TEST()
129 core::RefCountPtr<EagerContext> ctx(new EagerContext( in TEST()
147 core::RefCountPtr<EagerContext> ctx(new EagerContext( in TEST()
|
| D | eager_op_rewrite_registry_test.cc | 34 new tensorflow::EagerOperation(&orig_op->EagerContext()); in Run()
53 tensorflow::EagerContext* ctx = new tensorflow::EagerContext( in TEST()
|
| D | tensor_handle.cc | 227 EagerContext* ctx) { in CreateLocalHandle()
234 EagerContext* ctx) { in CreateLocalHandle()
243 Device* resource_device, EagerContext* ctx) in TensorHandle()
259 EagerContext* ctx) in TensorHandle()
281 EagerContext* ctx) { in CreateEmptyLocalHandle()
287 EagerContext* ctx) in TensorHandle()
305 EagerContext* ctx, in CreatePackedHandle()
334 EagerContext* ctx, in CreatePackedHandle()
352 EagerContext* ctx) in TensorHandle()
370 tensorflow::DataType dtype, Device* d, EagerContext* ctx, in CreateUnshapedRemoteHandle()
[all …]
|
| D | eager_operation.h | 39 explicit EagerOperation(tensorflow::EagerContext* ctx) in EagerOperation()
149 tensorflow::EagerContext& EagerContext() const { return ctx_; } in EagerContext() function
196 if (step_id != EagerContext::kGlobalRendezvousId) { in SetStepId()
243 tensorflow::EagerContext& ctx_;
|
| /external/tensorflow/tensorflow/core/function/ |
| D | runtime_client.cc | 56 EagerContext& GlobalEagerContext() { in GlobalEagerContext()
57 static EagerContext* global_ctx = []() { in GlobalEagerContext()
64 return new EagerContext( in GlobalEagerContext()
77 EagerContext& GlobalPythonEagerContext() { in GlobalPythonEagerContext()
78 EagerContext* ctx = reinterpret_cast<EagerContext*>(GetCEagerContext()); in GlobalPythonEagerContext()
84 EagerContext& ctx = this->eager_ctx_; in GetFunctionProto()
156 EagerContext& ctx = this->eager_ctx_; in CallFunction()
|
| D | runtime_client.h | 48 EagerContext& GlobalPythonEagerContext();
52 EagerContext& GlobalEagerContext();
60 explicit Runtime(EagerContext& eager_ctx) : eager_ctx_(eager_ctx) {} in Runtime()
78 EagerContext& eager_ctx_;
|
| D | runtime_client_pybind.cc | 26 PYBIND11_MAKE_OPAQUE(tensorflow::EagerContext);
29 pybind11::class_<tensorflow::EagerContext, tensorflow::EagerContextPtr> in PYBIND11_MODULE()
30 EagerContext(m, "EagerContext"); in PYBIND11_MODULE() local
40 .def(pybind11::init<tensorflow::EagerContext&>()) in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.h | 87 EagerContext* context);
117 static ServerContext* CreateMasterContext(tensorflow::EagerContext* ctx, in CreateMasterContext()
122 explicit ServerContext(tensorflow::EagerContext* ctx,
142 tensorflow::EagerContext* Context() const { return ctx_; } in Context()
158 tensorflow::EagerContext* ctx_;
212 EagerContext* eager_context, EagerExecutor* eager_executor,
215 EagerContext* eager_context);
217 EagerContext* eager_context);
219 EagerContext* eager_context);
|
| D | cluster_function_library_runtime.h | 40 EagerClusterFunctionLibraryRuntime(const uint64 context_id, EagerContext* ctx, in EagerClusterFunctionLibraryRuntime()
85 EagerContext* ctx_;
110 const uint64 context_id, EagerContext* ctx, WorkerSession* worker_session);
|
| D | eager_service_impl.cc | 56 Status GetNumRetvals(tensorflow::EagerContext* context, const string& op_name, in GetNumRetvals()
94 EagerContext* eager_context, in GetEagerOperationAndNumRetvals()
173 EagerContext* eager_context, int op_id, int num_retvals, in AddOpRetvalsToResponse()
282 tensorflow::EagerContext* ctx = new tensorflow::EagerContext( in CreateContext()
377 tensorflow::EagerContext* ctx = server_context->Context(); in UpdateContext()
445 const tensorflow::uint64 context_id, EagerContext* context) { in CreateMasterContext()
482 EagerContext* eager_context = context->Context(); in RunComponentFunction()
552 EagerContext* eager_context, in ExecuteOp()
663 tensorflow::EagerContext* ctx = context->Context(); in KeepAlive()
700 const RegisterFunctionOp& register_function, EagerContext* eager_context) { in RegisterFunction()
[all …]
|
| /external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| D | op_handler_selector.h | 24 class EagerContext; variable
36 using ::tensorflow::EagerContext;
44 EagerOpHandlerSelector(CoreRuntime* core_runtime, EagerContext* eager_context,
63 EagerContext* eager_context_;
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | tfrt_context.h | 28 class EagerContext; variable
87 tensorflow::EagerContext* GetEagerContext() { return eager_context_; } in GetEagerContext()
88 const tensorflow::EagerContext* GetEagerContext() const { in GetEagerContext()
108 tensorflow::EagerContext* eager_context_;
|
| D | transform_graph_function.h | 29 class EagerContext; variable
46 EagerContext* eager_ctx, bool enable_grappler,
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_data.h | 55 tensorflow::EagerContext* GetEagerContext() { return eager_context_; } in GetEagerContext()
79 tensorflow::EagerContext* eager_context_ = nullptr;
102 tensorflow::EagerContext* eager_context, TfLiteDelegate* flex_delegate);
|
| D | delegate_data_test.cc | 63 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TEST()
208 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TEST()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | kernel_utils.cc | 44 OwnedEagerContext owned_eager_context{new tensorflow::EagerContext( in InitEagerContext()
61 OwnedEagerContext owned_eager_context{new tensorflow::EagerContext( in InitEagerContext()
82 tfrt::Expected<EagerContext*> GetEagerContext(tfrt::ExecutionContext exec_ctx) { in GetEagerContext()
|
| D | kernel_utils.h | 52 using OwnedEagerContext = AutoReleasePtr<EagerContext>;
102 tfrt::Expected<EagerContext*> GetEagerContext(tfrt::ExecutionContext exec_ctx);
126 tfrt::Expected<EagerContext*> GetTFEagerContext() { in GetTFEagerContext()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | get_compiler_ir.h | 28 class EagerContext; variable
44 absl::string_view func_name, Device* dev, EagerContext* context,
|