Home
last modified time | relevance | path

Searched refs:TF_Operation (Results 1 – 25 of 27) 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,
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.h386 typedef struct TF_Operation TF_Operation; typedef
390 TF_Operation* oper;
396 TF_Operation* oper;
501 TF_Operation* input);
509 TF_Operation* op);
619 TF_CAPI_EXPORT extern TF_Operation* TF_FinishOperation(
625 TF_CAPI_EXPORT extern const char* TF_OperationName(TF_Operation* oper);
626 TF_CAPI_EXPORT extern const char* TF_OperationOpType(TF_Operation* oper);
627 TF_CAPI_EXPORT extern const char* TF_OperationDevice(TF_Operation* oper);
629 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 …]
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,
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()
53 void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device) { in SetRequestedDevice()
100 void RemoveAllControlInputs(TF_Graph* graph, TF_Operation* op) { in RemoveAllControlInputs()
172 void AddWhileInputHack(TF_Graph* graph, TF_Output new_src, TF_Operation* dst, in AddWhileInputHack()
Dc_api.cc684 void RecordMutation(TF_Graph* graph, const TF_Operation& op, in RecordMutation()
1063 TF_Operation* ToOperation(Node* node) { in ToOperation()
1064 return static_cast<TF_Operation*>(static_cast<void*>(node)); in ToOperation()
1071 const tensorflow::AttrValue* GetAttrValue(TF_Operation* oper, in GetAttrValue()
1229 void TF_AddControlInput(TF_OperationDescription* desc, TF_Operation* input) { in TF_AddControlInput()
1233 void TF_ColocateWith(TF_OperationDescription* desc, TF_Operation* op) { in TF_ColocateWith()
1457 static TF_Operation* TF_FinishOperationLocked(TF_OperationDescription* desc, in TF_FinishOperationLocked()
1492 TF_Operation* TF_FinishOperation(TF_OperationDescription* desc, in TF_FinishOperation()
1501 const char* TF_OperationName(TF_Operation* oper) { in TF_OperationName()
1505 const char* TF_OperationOpType(TF_Operation* oper) { in TF_OperationOpType()
[all …]
Dc_api_experimental.h101 TF_CAPI_EXPORT extern TF_Operation* TF_MakeFakeIteratorGetNextWithDatasets(
112 TF_CAPI_EXPORT extern TF_Operation* TF_MakeFileBasedIteratorGetNextWithDatasets(
299 TF_CAPI_EXPORT extern TF_Operation* TFE_AddEagerOpToGraph(
Dc_api_internal.h124 struct TF_Operation { struct
154 std::vector<TF_Operation*> return_nodes;
228 void RecordMutation(TF_Graph* graph, const TF_Operation& op,
Dc_api_experimental_test.cc34 TF_Operation* get_next = TF_MakeFakeIteratorGetNextWithDatasets(graph, s); in TestFakeIteratorStack()
76 TF_Operation* get_next = TF_MakeFileBasedIteratorGetNextWithDatasets( in TEST()
348 TF_Operation* graph_op = in AddEagerOpToGraphAndCheck()
404 AddEagerOpToGraphAndCheck(op, [this, &axis](TF_Operation* graph_op) { in TEST_F()
433 squeeze, [this, &boundaries](TF_Operation* squeeze_graph_op) { in TEST_F()
460 identityn, [this, kNumInputs](TF_Operation* graph_op) { in TEST_F()
Dc_api_function.cc521 const TF_Operation* const* opers, in ComputeBodyNodes()
563 const TF_Operation* const* opers, int ninputs, const TF_Output* inputs, in TF_GraphToFunctionWithControlOutputs()
565 int ncontrol_outputs, const TF_Operation* const* control_outputs, in TF_GraphToFunctionWithControlOutputs()
629 int num_opers, const TF_Operation* const* opers, in TF_GraphToFunction()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.h153 const std::vector<TF_Operation*>& targets,
169 const std::vector<TF_Operation*>& targets,
190 std::vector<TF_Output> GetOperationInputs(TF_Operation* oper);
193 std::vector<TF_Operation*> TF_OperationGetControlInputs_wrapper(
194 TF_Operation* oper);
197 std::vector<TF_Operation*> TF_OperationGetControlOutputs_wrapper(
198 TF_Operation* oper);
209 const std::vector<TF_Operation*>* opers,
212 const std::vector<TF_Operation*>* control_outputs,
Dtf_session.i65 PyObject* CreateWrappedTFOperation(TF_Operation* tf_operation) { in CreateWrappedTFOperation()
154 %typemap(out) std::vector<TF_Operation*> tensorflow::TF_OperationGetControlInputs_wrapper {
173 %typemap(out) std::vector<TF_Operation*> tensorflow::TF_OperationGetControlOutputs_wrapper {
410 %typemap(in, numinputs=0) (int* num_opers, TF_Operation*** opers)
411 (int num_opers, TF_Operation** opers) {
416 %typemap(argout) (int* num_opers, TF_Operation*** opers) {
422 TF_Operation** opers = *$2;
587 %typemap(in) (const std::vector<TF_Operation*>* opers)
588 (std::vector<TF_Operation*> opers) {
596 TF_Operation* oper_ptr;
[all …]
Dtf_session_helper.cc361 const std::vector<TF_Operation*>& targets, in TF_SessionRun_wrapper_helper()
447 const std::vector<TF_Operation*>& targets, in TF_SessionRun_wrapper()
522 const std::vector<TF_Operation*>& targets, in TF_SessionPRunSetup_wrapper()
539 const std::vector<TF_Operation*> targets; in TF_SessionPRun_wrapper()
550 std::vector<TF_Output> GetOperationInputs(TF_Operation* oper) { in GetOperationInputs()
559 std::vector<TF_Operation*> TF_OperationGetControlInputs_wrapper( in TF_OperationGetControlInputs_wrapper()
560 TF_Operation* oper) { in TF_OperationGetControlInputs_wrapper()
561 std::vector<TF_Operation*> control_inputs(TF_OperationNumControlInputs(oper)); in TF_OperationGetControlInputs_wrapper()
567 std::vector<TF_Operation*> TF_OperationGetControlOutputs_wrapper( in TF_OperationGetControlOutputs_wrapper()
568 TF_Operation* oper) { in TF_OperationGetControlOutputs_wrapper()
[all …]
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/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()
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()
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()
Dsession_jni.cc149 std::unique_ptr<TF_Operation* []> targets(new TF_Operation*[ntargets]); in Java_org_tensorflow_Session_run()
Dutils_jni.cc44 dst[i] = TF_Output{reinterpret_cast<TF_Operation*>(op_handles[i]), in resolveOutputs()
/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/core/common_runtime/eager/
Dtensor_handle.h44 struct TF_Operation;
51 TF_Operation* oper;

12