Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/
Dc_test_util.h47 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,
65 TF_Operation* AddWithCtrlDependency(TF_Operation* l, TF_Operation* r,
66 TF_Graph* graph, TF_Operation* ctrl_op,
69 TF_Operation* Add(TF_Output l, TF_Output r, TF_Graph* graph, TF_Status* s,
72 TF_Operation* Min(TF_Operation* l, TF_Operation* r, TF_Graph* graph,
[all …]
Dc_api_function_test.cc78 void Run(const std::vector<std::pair<TF_Operation*, TF_Tensor*>>& inputs, in Run() argument
79 TF_Operation* output, int32_t expected_result) { in Run()
86 void RunT(const std::vector<std::pair<TF_Operation*, TF_Tensor*>>& inputs, in RunT() argument
111 void Run(const std::vector<std::pair<TF_Operation*, TF_Tensor*>>& inputs, in Run() argument
143 std::vector<TF_Output> ToOutput(const std::vector<TF_Operation*> ops) { in ToOutput()
151 void Define(int num_opers, const std::vector<TF_Operation*>& opers, in Define()
152 const std::vector<TF_Operation*>& inputs, in Define()
153 const std::vector<TF_Operation*>& outputs, in Define()
175 void DefineT(int num_opers, const std::vector<TF_Operation*>& opers, in DefineT()
199 TF_Operation* Use(const std::vector<TF_Operation*>& inputs) { in Use()
[all …]
Dc_test_util.cc85 TF_Operation** op) { in PlaceholderHelper()
93 TF_Operation* Placeholder(TF_Graph* graph, TF_Status* s, const char* name) { in Placeholder()
94 TF_Operation* op; in Placeholder()
100 TF_Operation** op) { in ConstHelper()
110 TF_Operation* Const(TF_Tensor* t, TF_Graph* graph, TF_Status* s, in Const()
112 TF_Operation* op; in Const()
117 TF_Operation* ScalarConst(int32_t v, TF_Graph* graph, TF_Status* s, in ScalarConst()
123 TF_Operation* ScalarConst(double v, TF_Graph* graph, TF_Status* s, in ScalarConst()
129 void AddOpHelper(TF_Operation* l, TF_Operation* r, TF_Graph* graph, in AddOpHelper()
130 TF_Status* s, const char* name, TF_Operation** op, in AddOpHelper()
[all …]
Dc_api.h352 typedef struct TF_Operation TF_Operation; typedef
356 TF_Operation* oper;
362 TF_Operation* oper;
467 TF_Operation* input);
475 TF_Operation* op);
581 TF_CAPI_EXPORT extern TF_Operation* TF_FinishOperation(
587 TF_CAPI_EXPORT extern const char* TF_OperationName(TF_Operation* oper);
588 TF_CAPI_EXPORT extern const char* TF_OperationOpType(TF_Operation* oper);
589 TF_CAPI_EXPORT extern const char* TF_OperationDevice(TF_Operation* oper);
591 TF_CAPI_EXPORT extern int TF_OperationNumOutputs(TF_Operation* oper);
[all …]
Dwhile_loop_test.cc41 TF_Operation* placeholder = Placeholder( in Init()
82 std::vector<std::pair<TF_Operation*, TF_Tensor*>> inputs(inputs_.size()); in Run()
108 TF_Operation* one = ScalarConst(1, params_->cond_graph, s_); in CreateCondGraph()
109 TF_Operation* less_than = in CreateCondGraph()
158 TF_Operation* less_than = in TEST_F()
164 TF_Operation* add1 = Add(params_->body_inputs[0], params_->body_inputs[1], in TEST_F()
167 TF_Operation* one = ScalarConst(1, params_->body_graph, s_); in TEST_F()
169 TF_Operation* add2 = Add(add1, one, params_->body_graph, s_, "add2"); in TEST_F()
227 TF_Operation* six = ScalarConst(6, params_->cond_graph, s_); in TEST_F()
229 TF_Operation* less_than = in TEST_F()
[all …]
Dc_api_test.cc342 TF_Operation* feed = Placeholder(graph, s); in TEST()
420 TF_Operation* three = ScalarConst(3, graph, s); in TEST()
440 TF_Operation* feed = Placeholder(graph, s); in TEST()
469 TF_Operation* three = ScalarConst(3, graph, s); in TEST()
473 TF_Operation* add = Add(feed, three, graph, s); in TEST()
546 TF_Operation* neg = Neg(add, graph, s); in TEST()
566 TF_Operation* neg2 = TF_GraphOperationByName(graph, "neg"); in TEST()
572 TF_Operation* feed2 = TF_GraphOperationByName(graph, "feed"); in TEST()
584 TF_Operation* oper; in TEST()
643 TF_Operation* oper = ScalarConst(3, graph, s); in TEST()
[all …]
Dpython_api.h26 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);
Dpython_api.cc22 void AddControlInput(TF_Graph* graph, TF_Operation* op, TF_Operation* input) { in AddControlInput()
28 void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name, in SetAttr()
43 void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device) { in SetRequestedDevice()
90 void RemoveAllControlInputs(TF_Graph* graph, TF_Operation* op) { in RemoveAllControlInputs()
Dc_api.cc640 void RecordMutation(TF_Graph* graph, const TF_Operation& op, in RecordMutation()
960 TF_Operation* ToOperation(Node* node) { in ToOperation()
961 return static_cast<TF_Operation*>(static_cast<void*>(node)); in ToOperation()
968 const tensorflow::AttrValue* GetAttrValue(TF_Operation* oper, in GetAttrValue()
1126 void TF_AddControlInput(TF_OperationDescription* desc, TF_Operation* input) { in TF_AddControlInput()
1130 void TF_ColocateWith(TF_OperationDescription* desc, TF_Operation* op) { in TF_ColocateWith()
1347 static TF_Operation* TF_FinishOperationLocked(TF_OperationDescription* desc, in TF_FinishOperationLocked()
1382 TF_Operation* TF_FinishOperation(TF_OperationDescription* desc, in TF_FinishOperation()
1391 const char* TF_OperationName(TF_Operation* oper) { in TF_OperationName()
1395 const char* TF_OperationOpType(TF_Operation* oper) { in TF_OperationOpType()
[all …]
Dc_api_internal.h119 struct TF_Operation { struct
148 std::vector<TF_Operation*> return_nodes;
212 void RecordMutation(TF_Graph* graph, const TF_Operation& op,
Dc_api_function.cc448 const TF_Operation* const* opers, in ComputeBodyNodes()
489 int num_opers, const TF_Operation* const* opers, in TF_GraphToFunction()
/external/tensorflow/tensorflow/contrib/tensorrt/segment/
Dsegment_test.cc34 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) { in PlaceholderHelper()
79 TF_Operation* SegmentTest::Placeholder(TF_Graph* graph, TF_Status* s, in Placeholder()
81 TF_Operation* op; in Placeholder()
86 void SegmentTest::AddHelper(TF_Operation* l, TF_Operation* r, TF_Graph* graph, in AddHelper()
87 TF_Status* s, const char* name, TF_Operation** op, in AddHelper()
[all …]
/external/tensorflow/tensorflow/java/src/main/native/
Doperation_jni.cc36 TF_Operation* requireHandle(JNIEnv* env, jlong handle) { in requireHandle()
37 return requireHandleImpl<TF_Operation>(env, handle); in requireHandle()
48 TF_Operation* op = requireHandle(env, handle); in Java_org_tensorflow_Operation_name()
56 TF_Operation* op = requireHandle(env, handle); in Java_org_tensorflow_Operation_type()
64 TF_Operation* op = requireHandle(env, handle); in Java_org_tensorflow_Operation_numOutputs()
73 TF_Operation* op = requireHandle(env, handle); in Java_org_tensorflow_Operation_outputListLength()
92 TF_Operation* op = requireHandle(env, op_handle); in Java_org_tensorflow_Operation_shape()
145 TF_Operation* op = requireHandle(env, op_handle); in Java_org_tensorflow_Operation_dtype()
164 TF_Operation* op = requireHandle(env, handle); in Java_org_tensorflow_Operation_inputListLength()
Dsession_jni.cc82 dst[i] = TF_Output{reinterpret_cast<TF_Operation*>(op_handles[i]), in resolveOutputs()
177 std::unique_ptr<TF_Operation* []> targets(new TF_Operation*[ntargets]); in Java_org_tensorflow_Session_run()
Dgraph_jni.cc52 TF_Operation* op = TF_GraphOperationByName(g, cname); in Java_org_tensorflow_Graph_operation()
65 TF_Operation* operation = TF_GraphNextOperation(g, &pos); in Java_org_tensorflow_Graph_nextOperation()
Doperation_builder_jni.cc39 out->oper = reinterpret_cast<TF_Operation*>(op_handle); in resolveOutput()
77 TF_Operation* op = TF_FinishOperation(d, status); in Java_org_tensorflow_OperationBuilder_finish()
129 TF_Operation* control = reinterpret_cast<TF_Operation*>(op_handle); in Java_org_tensorflow_OperationBuilder_addControlInput()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.h126 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,
Dtf_session_helper.cc197 const std::vector<TF_Operation*>& targets, in TF_SessionRun_wrapper_helper()
283 const std::vector<TF_Operation*>& targets, in TF_SessionRun_wrapper()
358 const std::vector<TF_Operation*>& targets, in TF_SessionPRunSetup_wrapper()
375 const std::vector<TF_Operation*> targets; in TF_SessionPRun_wrapper()
386 std::vector<TF_Output> GetOperationInputs(TF_Operation* oper) { in GetOperationInputs()
395 std::vector<TF_Operation*> TF_OperationGetControlInputs_wrapper( in TF_OperationGetControlInputs_wrapper()
396 TF_Operation* oper) { in TF_OperationGetControlInputs_wrapper()
397 std::vector<TF_Operation*> control_inputs(TF_OperationNumControlInputs(oper)); in TF_OperationGetControlInputs_wrapper()
418 const std::vector<TF_Operation*>* opers, in TF_GraphToFunction_wrapper()
433 const TF_Operation* const* opers_array = nullptr; in TF_GraphToFunction_wrapper()
Dtf_session.i65 PyObject* CreateWrappedTFOperation(TF_Operation* tf_operation) { in CreateWrappedTFOperation()
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) {
398 TF_Operation** opers = *$2;
529 %typemap(in) (const std::vector<TF_Operation*>* opers)
530 (std::vector<TF_Operation*> opers) {
538 TF_Operation* oper_ptr;
540 $descriptor(TF_Operation*), 0);
Dtf_sessionrun_wrapper.i57 %typemap(in) (const std::vector<TF_Operation*>& targets)
58 (std::vector<TF_Operation*> targets) {
65 TF_Operation* oper_ptr;
/external/tensorflow/tensorflow/go/
Dsession.go205 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 { argument
Doperation.go27 c *C.TF_Operation
/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc637 TF_Operation* arg = TF_FinishOperation(arg_descr, status); in TEST()
643 TF_Operation* id = TF_FinishOperation(id_descr, status); in TEST()
699 TF_Operation* arg = TF_FinishOperation(arg_descr, status); in TEST()
705 TF_Operation* id = TF_FinishOperation(id_descr, status); in TEST()
/external/tensorflow/tensorflow/docs_src/extend/
Dlanguage_bindings.md82 `TF_Operation` in the C API.