Home
last modified time | relevance | path

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

12

/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.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);
201 void TF_GraphSetOutputHandleShapesAndTypes(TF_Graph* graph, TF_Output output,
208 void RecordMutation(TF_Graph* graph, const TF_Operation& op,
Dc_api.h171 typedef struct TF_Graph TF_Graph; typedef
174 TF_CAPI_EXPORT extern TF_Graph* TF_NewGraph(void);
178 TF_CAPI_EXPORT extern void TF_DeleteGraph(TF_Graph*);
224 TF_CAPI_EXPORT extern void TF_GraphSetTensorShape(TF_Graph* graph,
237 TF_CAPI_EXPORT extern int TF_GraphGetTensorNumDims(TF_Graph* graph,
253 TF_CAPI_EXPORT extern void TF_GraphGetTensorShape(TF_Graph* graph,
263 TF_Graph* graph, const char* op_type, const char* oper_name);
689 TF_Graph* graph, const char* oper_name);
697 TF_CAPI_EXPORT extern TF_Operation* TF_GraphNextOperation(TF_Graph* graph,
706 TF_CAPI_EXPORT extern void TF_GraphToGraphDef(TF_Graph* graph,
[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()
52 void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device) { in SetRequestedDevice()
58 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst, in UpdateEdge()
63 void RemoveAllControlInputs(TF_Graph* graph, TF_Operation* op) { in RemoveAllControlInputs()
75 void SetRequireShapeInferenceFns(TF_Graph* graph, bool require) { in SetRequireShapeInferenceFns()
85 std::string GetHandleShapeAndType(TF_Graph* graph, TF_Output output) { in GetHandleShapeAndType()
111 void SetHandleShapeAndType(TF_Graph* graph, TF_Output output, const void* proto, in SetHandleShapeAndType()
137 void AddWhileInputHack(TF_Graph* graph, TF_Output new_src, TF_Operation* dst, in AddWhileInputHack()
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.cc51 const TF_Graph* fn_body, const char* fn_name, int ninputs, in ProcessInputs()
87 Status ProcessOutputs(const TF_Graph* fn_body, const char* fn_name, in ProcessOutputs()
110 const TF_Graph* fn_body, const char* fn_name, int num_opers, in ComputeBodyNodes()
151 const TF_Graph* fn_body, const char* fn_name, in TF_GraphToFunctionWithControlOutputs()
225 TF_Function* TF_GraphToFunction(const TF_Graph* fn_body, const char* fn_name, in TF_GraphToFunction()
243 void TF_GraphCopyFunction(TF_Graph* g, const TF_Function* func, in TF_GraphCopyFunction()
266 int TF_GraphNumFunctions(TF_Graph* g) { in TF_GraphNumFunctions()
271 int TF_GraphGetFunctions(TF_Graph* g, TF_Function** funcs, int max_func, in TF_GraphGetFunctions()
Dc_api.cc241 void RecordMutation(TF_Graph* graph, const TF_Operation& op, in RecordMutation()
278 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 static 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()
1529 TF_Graph::TF_Graph() in TF_Graph() function in TF_Graph
1539 TF_Graph* TF_NewGraph() { return new TF_Graph; } in TF_NewGraph()
1541 void TF_DeleteGraph(TF_Graph* g) { in TF_DeleteGraph()
[all …]
Dc_api_test.cc99 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> graph(TF_NewGraph(), in GPUDeviceName()
364 TF_Graph* graph = TF_NewGraph(); in TEST()
461 TF_Graph* graph = TF_NewGraph(); in TEST()
639 TF_Graph* graph = TF_NewGraph(); in TEST()
695 TF_Graph* graph = TF_NewGraph(); in TEST()
872 TF_Graph* graph = TF_NewGraph(); in TEST()
923 TF_Graph* graph = TF_NewGraph(); in TEST()
979 TF_Graph* graph = TF_NewGraph(); in TEST()
1038 TF_Graph* graph = TF_NewGraph(); in RunMinTest()
1100 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.h106 TF_CAPI_EXPORT extern const char* TF_GraphDebugString(TF_Graph* graph,
Dwhile_loop_test.cc126 TF_Graph* graph_;
Dc_api_experimental.cc174 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.cc83 PYBIND11_MAKE_OPAQUE(TF_Graph);
102 py::class_<TF_Graph> TF_Graph_class(m, "TF_Graph"); in PYBIND11_MODULE()
190 [](const TF_Graph* fn_body, const char* fn_name, in PYBIND11_MODULE()
223 m.def("TF_GraphGetTensorShapeHelper", [](TF_Graph* graph, TF_Output output) { in PYBIND11_MODULE()
244 [](TF_Graph* graph, TF_Output output, const std::vector<int64_t>& dims, in PYBIND11_MODULE()
257 [](TF_Graph* graph, TF_Output output, const std::vector<int64_t>& dims, in PYBIND11_MODULE()
269 [](TF_Graph* graph, TF_Output output, in PYBIND11_MODULE()
315 [](TF_Graph* graph, py::handle& dtypes, const char* prefix) { in PYBIND11_MODULE()
326 [](TF_Graph* graph, const TF_SessionOptions* opts) { in PYBIND11_MODULE()
339 [](TF_Graph* graph, const TF_SessionOptions* opts) { in PYBIND11_MODULE()
[all …]
Dtf_session_helper.cc46 TF_Session* TF_NewSessionRef(TF_Graph* graph, const TF_SessionOptions* opts, in TF_NewSessionRef()
485 TF_Graph* graph, TF_Output output, TF_Status* out_status, in TF_GraphGetTensorShapeHelper()
579 const TF_Graph* fn_body, const char* fn_name, bool append_hash_to_fn_name, in TF_GraphToFunction_wrapper()
622 TF_Graph* graph, TF_Output output, in TF_GraphSetOutputHandleShapesAndTypes_wrapper()
635 void CreatePlaceholder(TF_Graph* graph, TF_Status* s, string&& name, in CreatePlaceholder()
645 std::vector<TF_Output> TF_CreatePlaceholders(TF_Graph* graph, PyObject* dtypes, in TF_CreatePlaceholders()
677 void TF_GraphSetTensorShape_wrapper(TF_Graph* graph, TF_Output output, in TF_GraphSetTensorShape_wrapper()
701 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.cc55 explicit GraphTensor(TF_Output output, TF_Graph* graph) in GraphTensor()
91 TF_Graph* graph_; // For shape inference.
97 explicit GraphOperation(TF_Graph* g) : TracingOperation(kGraph), g_(g) {} in GraphOperation()
313 TF_Graph* g_;
350 graph_(new TF_Graph(), TF_DeleteGraph), in GraphContext()
429 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> graph_;
Dc_api_experimental_test.cc246 TF_Graph* function_graph = TF_NewGraph(); in TEST()
/external/llvm-project/llvm/lib/Analysis/
DTFUtils.cpp35 using TFGraphPtr = std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)>;
/external/tensorflow/tensorflow/go/
Dgraph.go53 c *C.TF_Graph

12