| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| 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 | context.cc | 76 EagerContext::EagerContext( in EagerContext() function in tensorflow::EagerContext
130 AbstractTensorInterface* EagerContext::CreateInt64Scalar(int64 value) { in CreateInt64Scalar()
134 AbstractTensorInterface* EagerContext::CreateUint64Scalar(uint64 value) { in CreateUint64Scalar()
138 AbstractTensorInterface* EagerContext::CreateInt32Scalar(int32 value) { in CreateInt32Scalar()
142 AbstractTensorInterface* EagerContext::CreateFloatScalar(float value) { in CreateFloatScalar()
146 AbstractTensorInterface* EagerContext::CreateDoubleScalar(double value) { in CreateDoubleScalar()
150 AbstractTensorInterface* EagerContext::CreateHalfScalar(Eigen::half value) { in CreateHalfScalar()
154 AbstractTensorInterface* EagerContext::CreateStringScalar(tstring value) { in CreateStringScalar()
158 AbstractTensorInterface* EagerContext::CreateComplex128Scalar( in CreateComplex128Scalar()
163 AbstractTensorInterface* EagerContext::CreateBoolScalar(bool value) { in CreateBoolScalar()
[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,
95 EagerContext* ctx, const absl::InlinedVector<TensorHandle*, 4>& inputs, in ExecuteNode()
138 EagerContext* ctx_;
151 EagerContext* ctx, const absl::InlinedVector<TensorHandle*, 4>& inputs, in AsyncExecuteNode()
233 EagerContext* ctx_;
|
| D | context_distributed_manager.h | 26 class EagerContext; variable
32 explicit EagerContextDistributedManager(EagerContext* context) in EagerContextDistributedManager()
44 EagerContext* context_;
|
| D | execute.cc | 104 Status CopyInputToExpectedDevice(EagerContext* ctx, EagerOperation* op, in CopyInputToExpectedDevice()
183 EagerContext* ctx, EagerOperation* op, in ValidateInputTypeAndPlacement()
244 op->EagerContext().FuncLibDef()->Find(op->Name()); in GetOutputDTypes()
268 Status GetDeviceForInput(const EagerContext& ctx, TensorHandle* tensor_handle, in GetDeviceForInput()
324 Status GetFuncAttr(const EagerOperation* op, const EagerContext& ctx, in GetFuncAttr()
348 Status MustCompileWithXLA(const EagerOperation* op, const EagerContext& ctx, in MustCompileWithXLA()
386 EagerContext& ctx = op->EagerContext(); in GetOrCreateKernelAndDevice()
578 EagerContext* ctx, TensorHandle** output) { in CreateUnshapedOutput()
612 EagerContext& ctx = op->EagerContext(); in AddOrExecuteNode()
703 EagerContext& ctx = op->EagerContext(); in EagerLocalExecute()
[all …]
|
| D | core.cc | 106 ImmediateExecutionTensorHandle* EagerContext::CopyTensorHandleToDevice( in CopyTensorHandleToDevice()
155 ImmediateExecutionTensorHandle* EagerContext::CreateLocalHandle( in CreateLocalHandle()
162 ImmediateExecutionTensorHandle* EagerContext::CreateLocalHandleFromTFTensor( in CreateLocalHandleFromTFTensor()
175 ImmediateExecutionTensorHandle* EagerContext::TFTensorHandleFromInterface( in TFTensorHandleFromInterface()
183 ImmediateExecutionOperation* EagerContext::CreateOperation() { in CreateOperation()
187 Status EagerContext::RegisterFunction(AbstractFunction* f) { in RegisterFunction()
|
| 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 | 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 | 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 | custom_device_test.cc | 89 core::RefCountPtr<EagerContext> ctx(new EagerContext( in TEST()
114 core::RefCountPtr<EagerContext> ctx(new EagerContext( in TEST()
|
| D | execute.h | 51 EagerContext* ctx, const absl::InlinedVector<TensorHandle*, 4>& op_inputs,
63 Status EagerCopyToDevice(TensorHandle* h, EagerContext* ctx,
|
| D | placement_utils.cc | 51 static Status ValidateTensorHandleRemoteDevice(EagerContext* ctx, in ValidateTensorHandleRemoteDevice()
141 EagerContext& ctx = op.EagerContext(); in MaybePinToResourceDevice()
|
| D | execute_node.cc | 22 bool ExecuteNodeArgs::IsRemote(EagerContext* ctx, Device* input_device, in IsRemote()
36 Status ExecuteNodeArgs::InitPackedHandle(const int index, EagerContext* ctx, in InitPackedHandle()
65 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs, in Init()
|
| D | eager_operation.h | 39 explicit EagerOperation(tensorflow::EagerContext* ctx) in EagerOperation()
149 tensorflow::EagerContext& EagerContext() const { return ctx_; } in EagerContext() function
219 tensorflow::EagerContext& ctx_;
|
| D | placement_test.cc | 80 EagerContext* context() { return context_; } in context()
87 new EagerContext(opts, policy, in InitContext()
109 EagerContext* context_;
|
| D | copy_to_device_node.h | 30 const EagerContext& ctx, bool async, bool mirror) in CopyToDeviceNode()
85 const EagerContext& ctx_;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.h | 88 EagerContext* context);
118 static ServerContext* CreateMasterContext(tensorflow::EagerContext* ctx, in CreateMasterContext()
123 explicit ServerContext(tensorflow::EagerContext* ctx,
143 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 | 38 EagerClusterFunctionLibraryRuntime(const uint64 context_id, EagerContext* ctx, in EagerClusterFunctionLibraryRuntime()
83 EagerContext* ctx_;
108 const uint64 context_id, EagerContext* ctx, WorkerSession* worker_session);
|
| D | eager_service_impl.cc | 58 Status GetNumRetvals(tensorflow::EagerContext* context, const string& op_name, in GetNumRetvals()
96 EagerContext* eager_context, in GetEagerOperationAndNumRetvals()
173 EagerContext* eager_context, int op_id, int num_retvals, in AddOpRetvalsToResponse()
275 tensorflow::EagerContext* ctx = new tensorflow::EagerContext( in CreateContext()
341 tensorflow::EagerContext* ctx = server_context->Context(); in UpdateContext()
411 const tensorflow::uint64 context_id, EagerContext* context) { in CreateMasterContext()
448 EagerContext* eager_context = context->Context(); in RunComponentFunction()
512 EagerContext* eager_context, in ExecuteOp()
624 tensorflow::EagerContext* ctx = context->Context(); in KeepAlive()
661 const RegisterFunctionOp& register_function, EagerContext* eager_context) { in RegisterFunction()
[all …]
|
| D | remote_tensor_handle_data.h | 40 EagerContext* ctx);
80 EagerContext* ctx_;
|
| D | remote_mgr_test.cc | 32 TestRemoteMgr(bool is_master, EagerContext* ctx) in TestRemoteMgr()
54 ctx_ = new tensorflow::EagerContext( in RemoteMgrTest()
64 EagerContext* ctx_;
|
| /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;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | get_compiler_ir.h | 28 class EagerContext; variable
42 absl::string_view func_name, Device* dev, EagerContext* context,
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | variable.cc | 82 if (!tensorflow::isa<EagerContext>(ctx)) { in CreateUninitialized()
87 EagerContext* eager_ctx = reinterpret_cast<EagerContext*>(ctx); in CreateUninitialized()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | variable_ops_test.cc | 33 ImmediateTensorHandlePtr CreateScalarTensorHandle(EagerContext* context, in CreateScalarTensorHandle()
46 EagerContext* context() { return ctx_.get(); } in context()
|