Home
last modified time | relevance | path

Searched refs:TF_Graph (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/c/
Dc_test_util.h52 TF_Operation* Placeholder(TF_Graph* graph, TF_Status* s,
57 TF_Operation* Const(TF_Tensor* t, TF_Graph* graph, TF_Status* s,
60 TF_Operation* ScalarConst(bool v, TF_Graph* graph, TF_Status* s,
63 TF_Operation* ScalarConst(int32_t v, TF_Graph* graph, TF_Status* s,
66 TF_Operation* ScalarConst(double v, TF_Graph* graph, TF_Status* s,
69 TF_Operation* ScalarConst(float v, TF_Graph* graph, TF_Status* s,
72 TF_Operation* Add(TF_Operation* l, TF_Operation* r, TF_Graph* graph,
75 TF_Operation* AddNoCheck(TF_Operation* l, TF_Operation* r, TF_Graph* graph,
79 TF_Graph* graph, TF_Operation* ctrl_op,
82 TF_Operation* Add(TF_Output l, TF_Output r, TF_Graph* graph, TF_Status* s,
[all …]
Dpython_api.h28 void AddControlInput(TF_Graph* graph, TF_Operation* op, TF_Operation* input);
32 void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
37 void ClearAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
40 void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device);
43 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst,
46 void RemoveAllControlInputs(TF_Graph* graph, TF_Operation* op);
50 void SetRequireShapeInferenceFns(TF_Graph* graph, bool require);
65 std::string GetHandleShapeAndType(TF_Graph* graph, TF_Output output);
72 void SetHandleShapeAndType(TF_Graph* graph, TF_Output output, const void* proto,
78 void AddWhileInputHack(TF_Graph* graph, TF_Output new_src, TF_Operation* dst,
Dc_api_internal.h78 struct TF_Graph { struct
79 TF_Graph();
110 TF_Graph* parent; argument
115 TF_OperationDescription(TF_Graph* g, const char* op_type, in TF_OperationDescription() argument
120 TF_Graph* graph;
129 TF_Session(tensorflow::Session* s, TF_Graph* g);
132 TF_Graph* const graph;
134 tensorflow::mutex mu ACQUIRED_AFTER(TF_Graph::mu);
221 void TF_GraphSetOutputHandleShapesAndTypes(TF_Graph* graph, TF_Output output,
228 void RecordMutation(TF_Graph* graph, const TF_Operation& op,
Dc_api.h371 typedef struct TF_Graph TF_Graph; typedef
374 TF_CAPI_EXPORT extern TF_Graph* TF_NewGraph(void);
378 TF_CAPI_EXPORT extern void TF_DeleteGraph(TF_Graph*);
424 TF_CAPI_EXPORT extern void TF_GraphSetTensorShape(TF_Graph* graph,
437 TF_CAPI_EXPORT extern int TF_GraphGetTensorNumDims(TF_Graph* graph,
453 TF_CAPI_EXPORT extern void TF_GraphGetTensorShape(TF_Graph* graph,
463 TF_Graph* graph, const char* op_type, const char* oper_name);
893 TF_Graph* graph, const char* oper_name);
901 TF_CAPI_EXPORT extern TF_Operation* TF_GraphNextOperation(TF_Graph* graph,
910 TF_CAPI_EXPORT extern void TF_GraphToGraphDef(TF_Graph* graph,
[all …]
Dc_test_util.cc108 void PlaceholderHelper(TF_Graph* graph, TF_Status* s, const char* name, in PlaceholderHelper()
121 TF_Operation* Placeholder(TF_Graph* graph, TF_Status* s, const char* name, in Placeholder()
128 void ConstHelper(TF_Tensor* t, TF_Graph* graph, TF_Status* s, const char* name, in ConstHelper()
139 TF_Operation* Const(TF_Tensor* t, TF_Graph* graph, TF_Status* s, in Const()
146 TF_Operation* ScalarConst(bool v, TF_Graph* graph, TF_Status* s, in ScalarConst()
152 TF_Operation* ScalarConst(int32_t v, TF_Graph* graph, TF_Status* s, in ScalarConst()
158 TF_Operation* ScalarConst(double v, TF_Graph* graph, TF_Status* s, in ScalarConst()
164 TF_Operation* ScalarConst(float v, TF_Graph* graph, TF_Status* s, in ScalarConst()
170 void AddOpHelper(TF_Operation* l, TF_Operation* r, TF_Graph* graph, in AddOpHelper()
183 TF_Operation* Add(TF_Operation* l, TF_Operation* r, TF_Graph* graph, in Add()
[all …]
Dpython_api.cc23 void AddControlInput(TF_Graph* graph, TF_Operation* op, TF_Operation* input) { in AddControlInput()
29 void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name, in SetAttr()
44 void ClearAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name, in ClearAttr()
53 void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device) { in SetRequestedDevice()
59 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst, in UpdateEdge()
100 void RemoveAllControlInputs(TF_Graph* graph, TF_Operation* op) { in RemoveAllControlInputs()
112 void SetRequireShapeInferenceFns(TF_Graph* graph, bool require) { in SetRequireShapeInferenceFns()
122 std::string GetHandleShapeAndType(TF_Graph* graph, TF_Output output) { in GetHandleShapeAndType()
147 void SetHandleShapeAndType(TF_Graph* graph, TF_Output output, const void* proto, in SetHandleShapeAndType()
172 void AddWhileInputHack(TF_Graph* graph, TF_Output new_src, TF_Operation* dst, in AddWhileInputHack()
Dc_api_experimental.h84 TF_CAPI_EXPORT extern const char* TF_GraphDebugString(TF_Graph* graph,
102 TF_Graph* graph, TF_Status* status);
113 TF_Graph* graph, const char* file_path, int batch_size,
291 TF_CAPI_EXPORT extern TFE_TraceContext* TFE_NewTraceContext(TF_Graph* graph);
Dc_api.cc684 void RecordMutation(TF_Graph* graph, const TF_Operation& op, in RecordMutation()
721 void TF_GraphSetOutputHandleShapesAndTypes(TF_Graph* graph, TF_Output output, in TF_GraphSetOutputHandleShapesAndTypes()
1110 void TF_GraphSetTensorShape(TF_Graph* graph, TF_Output output, in TF_GraphSetTensorShape()
1128 int TF_GraphGetTensorNumDims(TF_Graph* graph, TF_Output output, in TF_GraphGetTensorNumDims()
1151 void TF_GraphGetTensorShape(TF_Graph* graph, TF_Output output, int64_t* dims, in TF_GraphGetTensorShape()
1198 static TF_OperationDescription* TF_NewOperationLocked(TF_Graph* graph, in TF_NewOperationLocked()
1205 TF_OperationDescription* TF_NewOperation(TF_Graph* graph, const char* op_type, in TF_NewOperation()
1939 TF_Graph::TF_Graph() in TF_Graph() function in TF_Graph
1946 TF_Graph* TF_NewGraph() { return new TF_Graph; } in TF_NewGraph()
1948 void TF_DeleteGraph(TF_Graph* g) { in TF_DeleteGraph()
[all …]
Dc_api_function.cc461 const TF_Graph* fn_body, const char* fn_name, int ninputs, in ProcessInputs()
497 Status ProcessOutputs(const TF_Graph* fn_body, const char* fn_name, in ProcessOutputs()
520 const TF_Graph* fn_body, const char* fn_name, int num_opers, in ComputeBodyNodes()
561 const TF_Graph* fn_body, const char* fn_name, in TF_GraphToFunctionWithControlOutputs()
627 TF_Function* TF_GraphToFunction(const TF_Graph* fn_body, const char* fn_name, in TF_GraphToFunction()
645 void TF_GraphCopyFunction(TF_Graph* g, const TF_Function* func, in TF_GraphCopyFunction()
668 int TF_GraphNumFunctions(TF_Graph* g) { in TF_GraphNumFunctions()
673 int TF_GraphGetFunctions(TF_Graph* g, TF_Function** funcs, int max_func, in TF_GraphGetFunctions()
Dc_api_test.cc96 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> graph(TF_NewGraph(), in GPUDeviceName()
361 TF_Graph* graph = TF_NewGraph(); in TEST()
458 TF_Graph* graph = TF_NewGraph(); in TEST()
658 TF_Graph* graph = TF_NewGraph(); in TEST()
835 TF_Graph* graph = TF_NewGraph(); in TEST()
886 TF_Graph* graph = TF_NewGraph(); in TEST()
942 TF_Graph* graph = TF_NewGraph(); in TEST()
1001 TF_Graph* graph = TF_NewGraph(); in RunMinTest()
1063 TF_Graph* graph = TF_NewGraph(); in TEST()
1133 TF_Graph* graph = TF_NewGraph(); in TEST()
[all …]
Dc_api_experimental_test.cc32 TF_Graph* graph = TF_NewGraph(); in TestFakeIteratorStack()
70 TF_Graph* graph = TF_NewGraph(); in TEST()
361 TF_Graph* graph_;
Dc_api_function_test.cc391 TF_Graph* func_graph_;
392 TF_Graph* host_graph_;
1216 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> func_graph( in DefineFunction()
1239 void NodeWithPlaceholderAttrHelper(TF_Graph* graph, TF_Status* s, in NodeWithPlaceholderAttrHelper()
1250 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> func_graph( in TEST_F()
1556 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> func_graph( in DefineStatefulFunction()
Dwhile_loop_test.cc127 TF_Graph* graph_;
Dc_api_experimental.cc123 const char* TF_GraphDebugString(TF_Graph* graph, size_t* len) { in TF_GraphDebugString()
8306 TF_Graph* graph, TF_Status* status) { in AddDatasetFunctionAndIteratorNodesToGraph()
8365 TF_Operation* TF_MakeFakeIteratorGetNextWithDatasets(TF_Graph* graph, in TF_MakeFakeIteratorGetNextWithDatasets()
8390 TF_Graph* graph, const char* file_path, int batch_size, in TF_MakeFileBasedIteratorGetNextWithDatasets()
9002 TFE_TraceContext* TFE_NewTraceContext(TF_Graph* graph) { in TFE_NewTraceContext()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.h43 TF_Session* TF_NewSessionRef(TF_Graph* graph, const TF_SessionOptions* opts,
142 TF_Graph* graph, TF_Output output, TF_Status* status, bool* unknown_shape);
208 const TF_Graph* fn_body, const char* fn_name, bool append_hash_to_fn_name,
226 TF_Graph* graph, TF_Output output,
233 void TF_GraphSetTensorShape_wrapper(TF_Graph* graph, TF_Output output,
243 PyObject* TF_TryEvaluateConstant_wrapper(TF_Graph* graph, TF_Output output,
Dtf_session_helper.cc46 TF_Session* TF_NewSessionRef(TF_Graph* graph, const TF_SessionOptions* opts, in TF_NewSessionRef()
494 TF_Graph* graph, TF_Output output, TF_Status* out_status, in TF_GraphGetTensorShapeHelper()
590 const TF_Graph* fn_body, const char* fn_name, bool append_hash_to_fn_name, in TF_GraphToFunction_wrapper()
633 TF_Graph* graph, TF_Output output, in TF_GraphSetOutputHandleShapesAndTypes_wrapper()
646 void TF_GraphSetTensorShape_wrapper(TF_Graph* graph, TF_Output output, in TF_GraphSetTensorShape_wrapper()
670 PyObject* TF_TryEvaluateConstant_wrapper(TF_Graph* graph, TF_Output output, in TF_TryEvaluateConstant_wrapper()
/external/tensorflow/tensorflow/java/src/main/native/
Dgraph_jni.cc37 TF_Graph* requireHandle(JNIEnv* env, jlong handle) { in requireHandle()
38 return requireHandleImpl<TF_Graph>(env, handle); in requireHandle()
53 TF_DeleteGraph(reinterpret_cast<TF_Graph*>(handle)); in Java_org_tensorflow_Graph_delete()
60 TF_Graph* g = requireHandle(env, handle); in Java_org_tensorflow_Graph_operation()
70 TF_Graph* g = requireHandle(env, handle); in Java_org_tensorflow_Graph_nextOperation()
89 TF_Graph* g = requireHandle(env, handle); in Java_org_tensorflow_Graph_importGraphDef()
119 TF_Graph* g = requireHandle(env, handle); in Java_org_tensorflow_Graph_toGraphDef()
147 TF_Graph* g = requireHandle(env, handle); in Java_org_tensorflow_Graph_addGradients()
202 TF_Graph* const subgraph, in buildSubgraph()
260 TF_Graph* g = requireHandle(env, handle); in Java_org_tensorflow_Graph_whileLoop()
Doperation_jni.cc40 TF_Graph* requireGraphHandle(JNIEnv* env, jlong handle) { in requireGraphHandle()
41 return requireHandleImpl<TF_Graph>(env, handle); in requireGraphHandle()
90 TF_Graph* graph = requireGraphHandle(env, graph_handle); in Java_org_tensorflow_Operation_shape()
143 TF_Graph* graph = requireGraphHandle(env, graph_handle); in Java_org_tensorflow_Operation_dtype()
Dsession_jni.cc86 TF_Graph* graph = reinterpret_cast<TF_Graph*>(graph_handle); in Java_org_tensorflow_Session_allocate2()
Dsaved_model_bundle_jni.cc65 TF_Graph* graph = TF_NewGraph(); in Java_org_tensorflow_SavedModelBundle_load()
Doperation_builder_jni.cc61 TF_Graph* graph = reinterpret_cast<TF_Graph*>(graph_handle); in Java_org_tensorflow_OperationBuilder_allocate()
/external/tensorflow/tensorflow/c/eager/
Dc_api_internal.h142 TF_Graph* const graph;
151 TFE_TraceContext(TF_Graph* graph) : graph(graph) {} in TFE_TraceContext()
Dc_api_test.cc1169 TF_Graph* function_graph = TF_NewGraph(); in TEST()
1236 TF_Graph* function_graph = TF_NewGraph(); in TEST()
/external/tensorflow/tensorflow/go/
Dgraph.go53 c *C.TF_Graph