Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/c/
Dpython_api.h29 void AddControlInput(TF_Graph* graph, TF_Operation* op, TF_Operation* input);
33 void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
38 void ClearAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
42 void SetFullType(TF_Graph* graph, TF_Operation* op,
45 void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device);
48 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst,
51 void RemoveAllControlInputs(TF_Graph* graph, TF_Operation* op);
55 void SetRequireShapeInferenceFns(TF_Graph* graph, bool require);
70 std::string GetHandleShapeAndType(TF_Graph* graph, TF_Output output);
77 void SetHandleShapeAndType(TF_Graph* graph, TF_Output output, const void* proto,
[all …]
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 …]
Dc_api_internal.h70 struct TF_Graph { struct
71 TF_Graph();
102 TF_Graph* parent; argument
107 TF_OperationDescription(TF_Graph* g, const char* op_type, in TF_OperationDescription() argument
112 TF_Graph* graph;
121 TF_Session(tensorflow::Session* s, TF_Graph* g);
124 TF_Graph* const graph;
126 tensorflow::mutex mu TF_ACQUIRED_AFTER(TF_Graph::mu);
198 void TF_GraphSetOutputHandleShapesAndTypes(TF_Graph* graph, TF_Output output,
205 void RecordMutation(TF_Graph* graph, const TF_Operation& op,
Dc_api.h154 typedef struct TF_Graph TF_Graph; typedef
157 TF_CAPI_EXPORT extern TF_Graph* TF_NewGraph(void);
161 TF_CAPI_EXPORT extern void TF_DeleteGraph(TF_Graph*);
207 TF_CAPI_EXPORT extern void TF_GraphSetTensorShape(TF_Graph* graph,
220 TF_CAPI_EXPORT extern int TF_GraphGetTensorNumDims(TF_Graph* graph,
236 TF_CAPI_EXPORT extern void TF_GraphGetTensorShape(TF_Graph* graph,
248 TF_Graph* graph, const char* op_type, const char* oper_name);
255 TF_Graph* graph, const char* op_type, const char* oper_name);
704 TF_Graph* graph, const char* oper_name);
712 TF_CAPI_EXPORT extern TF_Operation* TF_GraphNextOperation(TF_Graph* graph,
[all …]
Dpython_api.cc24 void AddControlInput(TF_Graph* graph, TF_Operation* op, TF_Operation* input) { in AddControlInput()
30 void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name, in SetAttr()
45 void ClearAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name, in ClearAttr()
52 void SetFullType(TF_Graph* graph, TF_Operation* op, in SetFullType()
59 void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device) { in SetRequestedDevice()
65 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst, in UpdateEdge()
70 void RemoveAllControlInputs(TF_Graph* graph, TF_Operation* op) { in RemoveAllControlInputs()
82 void SetRequireShapeInferenceFns(TF_Graph* graph, bool require) { in SetRequireShapeInferenceFns()
92 std::string GetHandleShapeAndType(TF_Graph* graph, TF_Output output) { in GetHandleShapeAndType()
118 void SetHandleShapeAndType(TF_Graph* graph, TF_Output output, const void* proto, in SetHandleShapeAndType()
[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 …]
Dc_api_function.cc52 const TF_Graph* fn_body, const char* fn_name, int ninputs, in ProcessInputs()
88 Status ProcessOutputs(const TF_Graph* fn_body, const char* fn_name, in ProcessOutputs()
111 const TF_Graph* fn_body, const char* fn_name, int num_opers, in ComputeBodyNodes()
152 const TF_Graph* fn_body, const char* fn_name, in TF_GraphToFunctionWithControlOutputs()
226 TF_Function* TF_GraphToFunction(const TF_Graph* fn_body, const char* fn_name, in TF_GraphToFunction()
244 void TF_GraphCopyFunction(TF_Graph* g, const TF_Function* func, in TF_GraphCopyFunction()
267 int TF_GraphNumFunctions(TF_Graph* g) { in TF_GraphNumFunctions()
272 int TF_GraphGetFunctions(TF_Graph* g, TF_Function** funcs, int max_func, in TF_GraphGetFunctions()
Dc_api.cc206 void RecordMutation(TF_Graph* graph, const TF_Operation& op, in RecordMutation()
243 void TF_GraphSetOutputHandleShapesAndTypes(TF_Graph* graph, TF_Output output, in TF_GraphSetOutputHandleShapesAndTypes()
688 void TF_GraphSetTensorShape(TF_Graph* graph, TF_Output output, in TF_GraphSetTensorShape()
706 int TF_GraphGetTensorNumDims(TF_Graph* graph, TF_Output output, in TF_GraphGetTensorNumDims()
729 void TF_GraphGetTensorShape(TF_Graph* graph, TF_Output output, int64_t* dims, in TF_GraphGetTensorShape()
776 TF_OperationDescription* TF_NewOperationLocked(TF_Graph* graph, in TF_NewOperationLocked()
783 TF_OperationDescription* TF_NewOperation(TF_Graph* graph, const char* op_type, in TF_NewOperation()
1554 TF_Graph::TF_Graph() in TF_Graph() function in TF_Graph
1564 TF_Graph* TF_NewGraph() { return new TF_Graph; } in TF_NewGraph()
1566 void TF_DeleteGraph(TF_Graph* g) { in TF_DeleteGraph()
[all …]
Dc_api_test.cc103 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> graph(TF_NewGraph(), in GPUDeviceName()
367 TF_Graph* graph = TF_NewGraph(); in TEST()
464 TF_Graph* graph = TF_NewGraph(); in TEST()
642 TF_Graph* graph = TF_NewGraph(); in TEST()
698 TF_Graph* graph = TF_NewGraph(); in TEST()
875 TF_Graph* graph = TF_NewGraph(); in TEST()
926 TF_Graph* graph = TF_NewGraph(); in TEST()
982 TF_Graph* graph = TF_NewGraph(); in TEST()
1041 TF_Graph* graph = TF_NewGraph(); in RunMinTest()
1103 TF_Graph* graph = TF_NewGraph(); in TEST()
[all …]
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()
1280 void NodeWithAttrHelper(TF_Graph* graph, TF_Status* s, const char* name, in NodeWithAttrHelper()
1292 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> func_graph( in TEST_F()
1594 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> func_graph( in DefineStatefulFunction()
Dc_api_experimental.h110 TF_CAPI_EXPORT extern const char* TF_GraphDebugString(TF_Graph* graph,
Dwhile_loop_test.cc126 TF_Graph* graph_;
Dc_api_experimental.cc183 const char* TF_GraphDebugString(TF_Graph* graph, size_t* len) { in TF_GraphDebugString()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.h43 TF_Session* TF_NewSessionRef(TF_Graph* graph, const TF_SessionOptions* opts,
141 TF_Graph* graph, TF_Output output, TF_Status* status, bool* unknown_shape);
207 const TF_Graph* fn_body, const char* fn_name, bool append_hash_to_fn_name,
225 TF_Graph* graph, TF_Output output,
234 std::vector<TF_Output> TF_CreatePlaceholders(TF_Graph* graph, PyObject* dtypes,
240 void TF_GraphSetTensorShape_wrapper(TF_Graph* graph, TF_Output output,
250 PyObject* TF_TryEvaluateConstant_wrapper(TF_Graph* graph, TF_Output output,
Dtf_session_wrapper.cc90 PYBIND11_MAKE_OPAQUE(TF_Graph);
109 py::class_<TF_Graph> TF_Graph_class(m, "TF_Graph"); in PYBIND11_MODULE()
197 [](const TF_Graph* fn_body, const char* fn_name, in PYBIND11_MODULE()
230 m.def("TF_GraphGetTensorShapeHelper", [](TF_Graph* graph, TF_Output output) { in PYBIND11_MODULE()
251 [](TF_Graph* graph, TF_Output output, const std::vector<int64_t>& dims, in PYBIND11_MODULE()
264 [](TF_Graph* graph, TF_Output output, const std::vector<int64_t>& dims, in PYBIND11_MODULE()
276 [](TF_Graph* graph, TF_Output output, in PYBIND11_MODULE()
322 [](TF_Graph* graph, py::handle& dtypes, const char* prefix) { in PYBIND11_MODULE()
333 [](TF_Graph* graph, const TF_SessionOptions* opts) { in PYBIND11_MODULE()
346 [](TF_Graph* graph, const TF_SessionOptions* opts) { in PYBIND11_MODULE()
[all …]
Dtf_session_helper.cc47 TF_Session* TF_NewSessionRef(TF_Graph* graph, const TF_SessionOptions* opts, in TF_NewSessionRef()
486 TF_Graph* graph, TF_Output output, TF_Status* out_status, in TF_GraphGetTensorShapeHelper()
580 const TF_Graph* fn_body, const char* fn_name, bool append_hash_to_fn_name, in TF_GraphToFunction_wrapper()
623 TF_Graph* graph, TF_Output output, in TF_GraphSetOutputHandleShapesAndTypes_wrapper()
636 void CreatePlaceholder(TF_Graph* graph, TF_Status* s, string&& name, in CreatePlaceholder()
646 std::vector<TF_Output> TF_CreatePlaceholders(TF_Graph* graph, PyObject* dtypes, in TF_CreatePlaceholders()
678 void TF_GraphSetTensorShape_wrapper(TF_Graph* graph, TF_Output output, in TF_GraphSetTensorShape_wrapper()
702 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()
Dgraph_operation_jni.cc39 TF_Graph* requireGraphHandle(JNIEnv* env, jlong handle) { in requireGraphHandle()
40 return requireHandleImpl<TF_Graph>(env, handle); in requireGraphHandle()
84 TF_Graph* graph = requireGraphHandle(env, graph_handle); in Java_org_tensorflow_GraphOperation_shape()
134 TF_Graph* graph = requireGraphHandle(env, graph_handle); in Java_org_tensorflow_GraphOperation_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()
Dgraph_operation_builder_jni.cc60 TF_Graph* graph = reinterpret_cast<TF_Graph*>(graph_handle); in Java_org_tensorflow_GraphOperationBuilder_allocate()
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental_graph.cc56 explicit GraphTensor(TF_Output output, TF_Graph* graph) in GraphTensor()
92 TF_Graph* graph_; // For shape inference.
98 explicit GraphOperation(TF_Graph* g) : TracingOperation(kGraph), g_(g) {} in GraphOperation()
305 TF_Graph* g_;
321 graph_(new TF_Graph(), TF_DeleteGraph), in GraphContext()
400 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> graph_;
Dc_api_experimental_test.cc247 TF_Graph* function_graph = TF_NewGraph(); in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
Dc_api_session_coordination_test.cc95 TF_Graph* graph = TF_NewGraph(); in TEST_P()
/external/tensorflow/tensorflow/go/
Dgraph.go53 c *C.TF_Graph

12