OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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