Home
last modified time | relevance | path

Searched refs:TFE_TraceContext (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/
Dc_api_experimental.h284 typedef struct TFE_TraceContext TFE_TraceContext; typedef
291 TF_CAPI_EXPORT extern TFE_TraceContext* TFE_NewTraceContext(TF_Graph* graph);
293 TF_CAPI_EXPORT extern void TFE_DeleteTraceContext(TFE_TraceContext* trace_ctx);
300 TFE_Op* op, TFE_TraceContext* trace_ctx, TFE_TensorHandle** retvals,
307 TFE_TraceContext* trace_ctx);
310 TFE_TraceContext* trace_ctx, unsigned int idx);
314 TFE_ConsumeInputConcreteTensorFromTraceContext(TFE_TraceContext* trace_ctx,
Dc_api_experimental.cc9002 TFE_TraceContext* TFE_NewTraceContext(TF_Graph* graph) { in TFE_NewTraceContext()
9003 return new TFE_TraceContext(graph); in TFE_NewTraceContext()
9006 void TFE_DeleteTraceContext(TFE_TraceContext* trace_ctx) { delete trace_ctx; } in TFE_DeleteTraceContext()
9010 static TF_Output getOrCreateSymbolicTensor(TFE_TraceContext* trace_ctx, in getOrCreateSymbolicTensor()
9056 TF_Operation* TFE_AddEagerOpToGraph(TFE_Op* op, TFE_TraceContext* trace_ctx, in TFE_AddEagerOpToGraph()
9117 int TFE_FinalizeInputTensorsFromTraceContext(TFE_TraceContext* trace_ctx) { in TFE_FinalizeInputTensorsFromTraceContext()
9130 TF_Output TFE_GetInputGraphNodeFromTraceContext(TFE_TraceContext* trace_ctx, in TFE_GetInputGraphNodeFromTraceContext()
9138 TFE_TraceContext* trace_ctx, unsigned int idx) { in TFE_ConsumeInputConcreteTensorFromTraceContext()
Dc_api_experimental_test.cc362 TFE_TraceContext* trace_ctx_;
/external/tensorflow/tensorflow/c/eager/
Dc_api_internal.h141 struct TFE_TraceContext { struct
151 TFE_TraceContext(TF_Graph* graph) : graph(graph) {} in TFE_TraceContext() function
153 ~TFE_TraceContext() { in ~TFE_TraceContext() argument