| /external/tensorflow/tensorflow/c/ |
| python_api.h | 26 void AddControlInput(TF_Graph* graph, TF_Operation* op, TF_Operation* input);
30 void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
33 void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device);
38 void RemoveAllControlInputs(TF_Graph* graph, TF_Operation* op);
|
| c_test_util.h | 47 TF_Operation* Placeholder(TF_Graph* graph, TF_Status* s,
50 TF_Operation* Const(TF_Tensor* t, TF_Graph* graph, TF_Status* s,
53 TF_Operation* ScalarConst(int32_t v, TF_Graph* graph, TF_Status* s,
56 TF_Operation* ScalarConst(double v, TF_Graph* graph, TF_Status* s,
59 TF_Operation* Add(TF_Operation* l, TF_Operation* r, TF_Graph* graph,
62 TF_Operation* AddNoCheck(TF_Operation* l, TF_Operation* r, TF_Graph* graph
[all...] |
| c_api_function_test.cc | 78 void Run(const std::vector<std::pair<TF_Operation*, TF_Tensor*>>& inputs,
79 TF_Operation* output, int32_t expected_result) {
86 void RunT(const std::vector<std::pair<TF_Operation*, TF_Tensor*>>& inputs,
111 void Run(const std::vector<std::pair<TF_Operation*, TF_Tensor*>>& inputs,
143 std::vector<TF_Output> ToOutput(const std::vector<TF_Operation*> ops) {
151 void Define(int num_opers, const std::vector<TF_Operation*>& opers,
152 const std::vector<TF_Operation*>& inputs,
153 const std::vector<TF_Operation*>& outputs,
175 void DefineT(int num_opers, const std::vector<TF_Operation*>& opers,
199 TF_Operation* Use(const std::vector<TF_Operation*>& inputs)
[all...] |
| c_test_util.cc | 85 TF_Operation** op) {
93 TF_Operation* Placeholder(TF_Graph* graph, TF_Status* s, const char* name) {
94 TF_Operation* op;
100 TF_Operation** op) {
110 TF_Operation* Const(TF_Tensor* t, TF_Graph* graph, TF_Status* s,
112 TF_Operation* op;
117 TF_Operation* ScalarConst(int32_t v, TF_Graph* graph, TF_Status* s,
123 TF_Operation* ScalarConst(double v, TF_Graph* graph, TF_Status* s,
129 void AddOpHelper(TF_Operation* l, TF_Operation* r, TF_Graph* graph
[all...] |
| c_api.h | 351 // invalidate old TF_Operation* pointers.
352 typedef struct TF_Operation TF_Operation;
356 TF_Operation* oper;
362 TF_Operation* oper;
467 TF_Operation* input);
475 TF_Operation* op);
573 // * a TF_Operation is added to the graph,
581 TF_CAPI_EXPORT extern TF_Operation* TF_FinishOperation(
584 // TF_Operation functions. Operations are immutable once created, s
[all...] |
| while_loop_test.cc | 41 TF_Operation* placeholder = Placeholder(
82 std::vector<std::pair<TF_Operation*, TF_Tensor*>> inputs(inputs_.size());
108 TF_Operation* one = ScalarConst(1, params_->cond_graph, s_);
109 TF_Operation* less_than =
158 TF_Operation* less_than =
164 TF_Operation* add1 = Add(params_->body_inputs[0], params_->body_inputs[1],
167 TF_Operation* one = ScalarConst(1, params_->body_graph, s_);
169 TF_Operation* add2 = Add(add1, one, params_->body_graph, s_, "add2");
227 TF_Operation* six = ScalarConst(6, params_->cond_graph, s_);
229 TF_Operation* less_than
[all...] |
| python_api.cc | 22 void AddControlInput(TF_Graph* graph, TF_Operation* op, TF_Operation* input) {
28 void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
43 void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device) {
90 void RemoveAllControlInputs(TF_Graph* graph, TF_Operation* op) {
|
| c_api_test.cc | 342 TF_Operation* feed = Placeholder(graph, s);
420 TF_Operation* three = ScalarConst(3, graph, s);
440 TF_Operation* feed = Placeholder(graph, s);
443 // Test TF_Operation*() query functions.
460 // Test not found errors in TF_Operation*() query functions.
469 TF_Operation* three = ScalarConst(3, graph, s);
473 TF_Operation* add = Add(feed, three, graph, s);
476 // Test TF_Operation*() query functions.
546 TF_Operation* neg = Neg(add, graph, s);
566 TF_Operation* neg2 = TF_GraphOperationByName(graph, "neg")
[all...] |
| c_api_internal.h | 119 struct TF_Operation {
148 std::vector<TF_Operation*> return_nodes;
212 void RecordMutation(TF_Graph* graph, const TF_Operation& op,
|
| c_api.cc | 640 void RecordMutation(TF_Graph* graph, const TF_Operation& op,
[all...] |
| c_api_function.cc | 448 const TF_Operation* const* opers,
489 int num_opers, const TF_Operation* const* opers,
|
| /external/tensorflow/tensorflow/contrib/tensorrt/segment/ |
| segment_test.cc | 34 TF_Operation* Placeholder(TF_Graph* graph, TF_Status* s, const char* name);
35 TF_Operation* Add(TF_Operation* l, TF_Operation* r, TF_Graph* graph,
43 TF_Operation** op);
44 void AddHelper(TF_Operation* l, TF_Operation* r, TF_Graph* graph,
45 TF_Status* s, const char* name, TF_Operation** op, bool check);
71 const char* name, TF_Operation** op) {
79 TF_Operation* SegmentTest::Placeholder(TF_Graph* graph, TF_Status* s
[all...] |
| /external/tensorflow/tensorflow/python/client/ |
| tf_session_helper.h | 126 const std::vector<TF_Operation*>& targets,
142 const std::vector<TF_Operation*>& targets,
164 std::vector<TF_Output> GetOperationInputs(TF_Operation* oper);
167 std::vector<TF_Operation*> TF_OperationGetControlInputs_wrapper(
168 TF_Operation* oper);
179 const std::vector<TF_Operation*>* opers,
|
| tf_sessionrun_wrapper.i | 57 %typemap(in) (const std::vector<TF_Operation*>& targets)
58 (std::vector<TF_Operation*> targets) {
65 TF_Operation* oper_ptr;
|
| tf_session_helper.cc | 197 const std::vector<TF_Operation*>& targets,
283 const std::vector<TF_Operation*>& targets,
358 const std::vector<TF_Operation*>& targets,
375 const std::vector<TF_Operation*> targets;
386 std::vector<TF_Output> GetOperationInputs(TF_Operation* oper) {
395 std::vector<TF_Operation*> TF_OperationGetControlInputs_wrapper(
396 TF_Operation* oper) {
397 std::vector<TF_Operation*> control_inputs(TF_OperationNumControlInputs(oper));
418 const std::vector<TF_Operation*>* opers,
433 const TF_Operation* const* opers_array = nullptr
[all...] |
| tf_session.i | 63 // Helper function to convert a TF_Operation to a wrapped TF_Operation Python
65 PyObject* CreateWrappedTFOperation(TF_Operation* tf_operation) {
67 return SWIG_NewPointerObj(tf_operation, SWIGTYPE_p_TF_Operation, 0);
148 // Build a Python list of TF_Operation* and return it.
149 %typemap(out) std::vector<TF_Operation*> tensorflow::TF_OperationGetControlInputs_wrapper {
386 %typemap(in, numinputs=0) (int* num_opers, TF_Operation*** opers)
387 (int num_opers, TF_Operation** opers) {
392 %typemap(argout) (int* num_opers, TF_Operation*** opers)
[all...] |
| /external/tensorflow/tensorflow/java/src/main/native/ |
| operation_jni.cc | 36 TF_Operation* requireHandle(JNIEnv* env, jlong handle) {
37 return requireHandleImpl<TF_Operation>(env, handle);
48 TF_Operation* op = requireHandle(env, handle);
56 TF_Operation* op = requireHandle(env, handle);
64 TF_Operation* op = requireHandle(env, handle);
73 TF_Operation* op = requireHandle(env, handle);
92 TF_Operation* op = requireHandle(env, op_handle);
145 TF_Operation* op = requireHandle(env, op_handle);
164 TF_Operation* op = requireHandle(env, handle);
|
| graph_jni.cc | 52 TF_Operation* op = TF_GraphOperationByName(g, cname);
65 TF_Operation* operation = TF_GraphNextOperation(g, &pos);
|
| session_jni.cc | 82 dst[i] = TF_Output{reinterpret_cast<TF_Operation*>(op_handles[i]),
177 std::unique_ptr<TF_Operation* []> targets(new TF_Operation*[ntargets]);
|
| operation_builder_jni.cc | 39 out->oper = reinterpret_cast<TF_Operation*>(op_handle);
77 TF_Operation* op = TF_FinishOperation(d, status);
129 TF_Operation* control = reinterpret_cast<TF_Operation*>(op_handle);
|
| /external/tensorflow/tensorflow/go/ |
| session.go | 205 ctargets = make([]*C.TF_Operation, len(targets))
209 pctargets **C.TF_Operation
346 targets []*C.TF_Operation
353 targets: make([]*C.TF_Operation, len(targets)),
390 func ptrOperation(l []*C.TF_Operation) **C.TF_Operation {
|
| operation.go | 27 c *C.TF_Operation
|
| /external/tensorflow/tensorflow/c/eager/ |
| c_api_test.cc | 637 TF_Operation* arg = TF_FinishOperation(arg_descr, status);
643 TF_Operation* id = TF_FinishOperation(id_descr, status);
699 TF_Operation* arg = TF_FinishOperation(arg_descr, status);
705 TF_Operation* id = TF_FinishOperation(id_descr, status);
[all...] |
