HomeSort by relevance Sort by last modified time
    Searched defs:TFE_Context (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/c/eager/
c_api.h 86 typedef struct TFE_Context TFE_Context;
88 TF_CAPI_EXPORT extern TFE_Context* TFE_NewContext(
90 TF_CAPI_EXPORT extern void TFE_DeleteContext(TFE_Context* ctx,
92 TF_CAPI_EXPORT extern TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx,
97 TF_CAPI_EXPORT extern void TFE_ContextClearCaches(TFE_Context* ctx);
104 TFE_Context*, TFE_ContextDevicePlacementPolicy);
109 TFE_ContextGetDevicePlacementPolicy(TFE_Context*);
137 TFE_TensorHandle* h, TFE_Context* ctx, const char* device_name,
153 TF_CAPI_EXPORT extern TFE_Op* TFE_NewOp(TFE_Context* ctx
    [all...]
c_api_internal.h 46 struct TFE_Context {
47 explicit TFE_Context(const TFE_ContextOptions& opts, TF_Session* s)
63 // TFE_Context is an extension of TF_Session. And TF_Session needs a TF_Graph.
102 // provided with the appropriate TFE_Context.
104 // TODO(ashankar): Reference count TFE_Context to ensure that 'd' of a
105 // TFE_TensorHandle does not outlive the TFE_Context from which it came?
112 TFE_Op(TFE_Context* ctx, const char* op, const tensorflow::AttrTypeMap* t)
117 TFE_Context* ctx; // Must outlive the TFE_Op.

Completed in 519 milliseconds