Home
last modified time | relevance | path

Searched refs:TF_Operation (Results 1 – 23 of 23) 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,
78 TF_Operation* AddWithCtrlDependency(TF_Operation* l, TF_Operation* r,
79 TF_Graph* graph, TF_Operation* ctrl_op,
[all …]
Dc_api_function_test.cc80 void Run(const std::vector<std::pair<TF_Operation*, TF_Tensor*>>& inputs, in Run() argument
81 TF_Operation* output, int32_t expected_result) { in Run()
88 void RunT(const std::vector<std::pair<TF_Operation*, TF_Tensor*>>& inputs, in RunT() argument
113 void Run(const std::vector<std::pair<TF_Operation*, TF_Tensor*>>& inputs, in Run() argument
145 std::vector<TF_Output> ToOutput(const std::vector<TF_Operation*> ops) { in ToOutput()
153 void Define(int num_opers, const std::vector<TF_Operation*>& opers, in Define()
154 const std::vector<TF_Operation*>& inputs, in Define()
155 const std::vector<TF_Operation*>& outputs, in Define()
177 void DefineT(int num_opers, const std::vector<TF_Operation*>& opers, in DefineT()
202 TF_Operation* Use(const std::vector<TF_Operation*>& inputs) { in Use()
[all …]
Dc_test_util.cc110 TF_Operation** op) { in PlaceholderHelper()
121 TF_Operation* Placeholder(TF_Graph* graph, TF_Status* s, const char* name, in Placeholder()
123 TF_Operation* op; in Placeholder()
129 TF_Operation** op) { in ConstHelper()
139 TF_Operation* Const(TF_Tensor* t, TF_Graph* graph, TF_Status* s, in Const()
141 TF_Operation* op; 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()
[all …]
Dc_api.h177 typedef struct TF_Operation TF_Operation; typedef
181 TF_Operation* oper;
187 TF_Operation* oper;
292 TF_Operation* input);
300 TF_Operation* op);
410 TF_CAPI_EXPORT extern TF_Operation* TF_FinishOperation(
416 TF_CAPI_EXPORT extern const char* TF_OperationName(TF_Operation* oper);
417 TF_CAPI_EXPORT extern const char* TF_OperationOpType(TF_Operation* oper);
418 TF_CAPI_EXPORT extern const char* TF_OperationDevice(TF_Operation* oper);
420 TF_CAPI_EXPORT extern int TF_OperationNumOutputs(TF_Operation* oper);
[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);
46 void RemoveAllControlInputs(TF_Graph* graph, TF_Operation* op);
78 void AddWhileInputHack(TF_Graph* graph, TF_Output new_src, TF_Operation* dst,
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.cc363 TF_Operation* feed = Placeholder(graph, s); in TEST()
441 TF_Operation* three = ScalarConst(3, graph, s); in TEST()
461 TF_Operation* feed = Placeholder(graph, s); in TEST()
490 TF_Operation* three = ScalarConst(3, graph, s); in TEST()
494 TF_Operation* add = Add(feed, three, graph, s); in TEST()
567 TF_Operation* neg = Neg(add, graph, s); in TEST()
587 TF_Operation* neg2 = TF_GraphOperationByName(graph, "neg"); in TEST()
593 TF_Operation* feed2 = TF_GraphOperationByName(graph, "feed"); in TEST()
605 TF_Operation* oper; in TEST()
664 TF_Operation* oper = ScalarConst(3, graph, s); in TEST()
[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()
99 void RemoveAllControlInputs(TF_Graph* graph, TF_Operation* op) { in RemoveAllControlInputs()
171 void AddWhileInputHack(TF_Graph* graph, TF_Output new_src, TF_Operation* dst, in AddWhileInputHack()
Dc_api.cc241 void RecordMutation(TF_Graph* graph, const TF_Operation& op, in RecordMutation()
639 TF_Operation* ToOperation(Node* node) { in ToOperation()
640 return static_cast<TF_Operation*>(static_cast<void*>(node)); in ToOperation()
647 const tensorflow::AttrValue* GetAttrValue(TF_Operation* oper, in GetAttrValue()
805 void TF_AddControlInput(TF_OperationDescription* desc, TF_Operation* input) { in TF_AddControlInput()
809 void TF_ColocateWith(TF_OperationDescription* desc, TF_Operation* op) { in TF_ColocateWith()
1033 static TF_Operation* TF_FinishOperationLocked(TF_OperationDescription* desc, in TF_FinishOperationLocked()
1069 TF_Operation* TF_FinishOperation(TF_OperationDescription* desc, in TF_FinishOperation()
1078 const char* TF_OperationName(TF_Operation* oper) { in TF_OperationName()
1082 const char* TF_OperationOpType(TF_Operation* oper) { in TF_OperationOpType()
[all …]
Dc_api_internal.h116 struct TF_Operation { struct
146 std::vector<TF_Operation*> return_nodes;
211 void RecordMutation(TF_Graph* graph, const TF_Operation& op,
Dc_api_function.cc111 const TF_Operation* const* opers, in ComputeBodyNodes()
153 const TF_Operation* const* opers, int ninputs, const TF_Output* inputs, in TF_GraphToFunctionWithControlOutputs()
155 int ncontrol_outputs, const TF_Operation* const* control_outputs, in TF_GraphToFunctionWithControlOutputs()
221 int num_opers, const TF_Operation* const* opers, in TF_GraphToFunction()
Dc_api_experimental.cc52 static TF_Operation* ToTF_Operation(Node* node) { in ToTF_Operation()
53 return static_cast<TF_Operation*>(static_cast<void*>(node)); in ToTF_Operation()
232 TF_Operation* dequeue_op = TF_GraphOperationByName( in TF_DequeueNamedTensor()
278 TF_Operation* enqueue_op = TF_GraphOperationByName( in TF_EnqueueNamedTensor()
287 TF_Operation* placeholder_op = TF_GraphOperationByName( in TF_EnqueueNamedTensor()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.h152 const std::vector<TF_Operation*>& targets,
168 const std::vector<TF_Operation*>& targets,
189 std::vector<TF_Output> GetOperationInputs(TF_Operation* oper);
192 std::vector<TF_Operation*> TF_OperationGetControlInputs_wrapper(
193 TF_Operation* oper);
196 std::vector<TF_Operation*> TF_OperationGetControlOutputs_wrapper(
197 TF_Operation* oper);
208 const std::vector<TF_Operation*>* opers,
211 const std::vector<TF_Operation*>* control_outputs,
Dtf_session_helper.cc356 const std::vector<TF_Operation*>& targets, in TF_SessionRun_wrapper_helper()
443 const std::vector<TF_Operation*>& targets, in TF_SessionRun_wrapper()
518 const std::vector<TF_Operation*>& targets, in TF_SessionPRunSetup_wrapper()
535 const std::vector<TF_Operation*> targets; in TF_SessionPRun_wrapper()
546 std::vector<TF_Output> GetOperationInputs(TF_Operation* oper) { in GetOperationInputs()
553 std::vector<TF_Operation*> TF_OperationGetControlInputs_wrapper( in TF_OperationGetControlInputs_wrapper()
554 TF_Operation* oper) { in TF_OperationGetControlInputs_wrapper()
555 std::vector<TF_Operation*> control_inputs(TF_OperationNumControlInputs(oper)); in TF_OperationGetControlInputs_wrapper()
561 std::vector<TF_Operation*> TF_OperationGetControlOutputs_wrapper( in TF_OperationGetControlOutputs_wrapper()
562 TF_Operation* oper) { in TF_OperationGetControlOutputs_wrapper()
[all …]
Dtf_session_wrapper.cc85 PYBIND11_MAKE_OPAQUE(TF_Operation);
103 py::class_<TF_Operation> TF_Operation_class(m, "TF_Operation"); in PYBIND11_MODULE()
192 absl::optional<std::vector<TF_Operation*>> opers_opt, in PYBIND11_MODULE()
196 const std::vector<TF_Operation*> control_outputs, in PYBIND11_MODULE()
415 const std::vector<TF_Operation*>& targets, in PYBIND11_MODULE()
519 const std::vector<TF_Operation*>& targets) { in PYBIND11_MODULE()
594 TF_Operation* output = TF_FinishOperation(desc, status.get()); in PYBIND11_MODULE()
601 [](TF_Operation* oper, const char* attr_name) { in PYBIND11_MODULE()
646 [](TF_Operation* oper, TF_Buffer* output_node_def) { in PYBIND11_MODULE()
654 [](TF_Operation* oper, const char* attr_name, in PYBIND11_MODULE()
[all …]
/external/tensorflow/tensorflow/java/src/main/native/
Dgraph_operation_jni.cc35 TF_Operation* requireHandle(JNIEnv* env, jlong handle) { in requireHandle()
36 return requireHandleImpl<TF_Operation>(env, handle); in requireHandle()
46 TF_Operation* op = requireHandle(env, handle); in Java_org_tensorflow_GraphOperation_name()
53 TF_Operation* op = requireHandle(env, handle); in Java_org_tensorflow_GraphOperation_type()
60 TF_Operation* op = requireHandle(env, handle); in Java_org_tensorflow_GraphOperation_numOutputs()
67 TF_Operation* op = requireHandle(env, handle); in Java_org_tensorflow_GraphOperation_outputListLength()
86 TF_Operation* op = requireHandle(env, op_handle); in Java_org_tensorflow_GraphOperation_shape()
137 TF_Operation* op = requireHandle(env, op_handle); in Java_org_tensorflow_GraphOperation_dtype()
154 TF_Operation* op = requireHandle(env, handle); in Java_org_tensorflow_GraphOperation_inputListLength()
Dgraph_jni.cc41 TF_Operation* requireOperationHandle(JNIEnv* env, jlong handle) { in requireOperationHandle()
42 return requireHandleImpl<TF_Operation>(env, handle); in requireOperationHandle()
63 TF_Operation* op = TF_GraphOperationByName(g, cname); in Java_org_tensorflow_Graph_operation()
74 TF_Operation* operation = TF_GraphNextOperation(g, &pos); in Java_org_tensorflow_Graph_nextOperation()
292 TF_Operation* cond_output_op = in Java_org_tensorflow_Graph_whileLoop()
303 TF_Operation* body_output_op = in Java_org_tensorflow_Graph_whileLoop()
Dgraph_operation_builder_jni.cc38 out->oper = reinterpret_cast<TF_Operation*>(op_handle); in resolveOutput()
76 TF_Operation* op = TF_FinishOperation(d, status); in Java_org_tensorflow_GraphOperationBuilder_finish()
131 TF_Operation* control = reinterpret_cast<TF_Operation*>(op_handle); in Java_org_tensorflow_GraphOperationBuilder_addControlInput()
Dutils_jni.cc44 dst[i] = TF_Output{reinterpret_cast<TF_Operation*>(op_handles[i]), in resolveOutputs()
Dsession_jni.cc149 std::unique_ptr<TF_Operation* []> targets(new TF_Operation*[ntargets]); in Java_org_tensorflow_Session_run()
/external/tensorflow/tensorflow/go/
Dsession.go218 ctargets = make([]*C.TF_Operation, len(targets))
222 pctargets **C.TF_Operation
359 targets []*C.TF_Operation
366 targets: make([]*C.TF_Operation, len(targets)),
403 func ptrOperation(l []*C.TF_Operation) **C.TF_Operation { argument
Doperation.go27 c *C.TF_Operation
/external/tensorflow/tensorflow/c/eager/
Dc_api_experimental_test.cc305 TF_Operation* arg = TF_FinishOperation(arg_descr, status); in TEST()
311 TF_Operation* id = TF_FinishOperation(id_descr, status); in TEST()
381 TF_Operation* arg = TF_FinishOperation(arg_descr, status); in TEST()
387 TF_Operation* id = TF_FinishOperation(id_descr, status); in TEST()