Home
last modified time | relevance | path

Searched refs:TF_Output (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.h114 TF_Graph* graph, TF_Output output, TF_Status* out_status,
123 const std::vector<TF_Output>& inputs,
125 const std::vector<TF_Output>& outputs,
140 const std::vector<TF_Output>& inputs,
141 const std::vector<TF_Output>& outputs,
157 const std::vector<TF_Output>& inputs,
159 const std::vector<TF_Output>& outputs,
164 std::vector<TF_Output> GetOperationInputs(TF_Operation* oper);
173 TF_Output oper_out);
180 const std::vector<TF_Output>& inputs, const std::vector<TF_Output>& outputs,
[all …]
Dtf_sessionrun_wrapper.i30 %typemap(in) (const std::vector<TF_Output>& inputs,
32 (std::vector<TF_Output> inputs, std::vector<PyObject*> input_ndarrays) {
40 TF_Output* input_ptr;
74 %typemap(in) (const std::vector<TF_Output>& outputs)
75 (std::vector<TF_Output> outputs) {
84 %typemap(in) (const std::vector<TF_Output>& inputs) =
85 (const std::vector<TF_Output>& outputs);
Dtf_session_helper.cc194 const std::vector<TF_Output>& inputs, in TF_SessionRun_wrapper_helper()
196 const std::vector<TF_Output>& outputs, in TF_SessionRun_wrapper_helper()
280 const std::vector<TF_Output>& inputs, in TF_SessionRun_wrapper()
282 const std::vector<TF_Output>& outputs, in TF_SessionRun_wrapper()
330 TF_Graph* graph, TF_Output output, TF_Status* out_status, in TF_GraphGetTensorShapeHelper()
356 const std::vector<TF_Output>& inputs, in TF_SessionPRunSetup_wrapper()
357 const std::vector<TF_Output>& outputs, in TF_SessionPRunSetup_wrapper()
370 const std::vector<TF_Output>& inputs, in TF_SessionPRun_wrapper()
372 const std::vector<TF_Output>& outputs, in TF_SessionPRun_wrapper()
386 std::vector<TF_Output> GetOperationInputs(TF_Operation* oper) { in GetOperationInputs()
[all …]
Dtf_session.i32 std::vector<TF_Output>* vec, in PyTensorListToVector()
41 TF_Output* input_ptr; in PyTensorListToVector()
54 PyObject* CreateWrappedTFOutput(TF_Output tf_output) { in CreateWrappedTFOutput()
57 TF_Output* tf_output_ptr = new TF_Output(tf_output); in CreateWrappedTFOutput()
185 %typemap(out) std::vector<TF_Output> tensorflow::GetOperationInputs {
192 const std::vector<TF_Output>& tf_outputs = $1;
356 %typemap(in) (const TF_Output* inputs, int num_inputs)
357 (std::vector<TF_Output> inputs) {
367 %typemap(in, numinputs=0) (int* num_outputs, TF_Output** outputs)
368 (int num_outputs, TF_Output* outputs) {
[all …]
/external/tensorflow/tensorflow/c/
Dc_api.h361 typedef struct TF_Output { struct
364 } TF_Output; argument
391 TF_Output output,
404 TF_Output output,
420 TF_Output output,
457 TF_Output input);
462 const TF_Output* inputs,
592 TF_CAPI_EXPORT extern TF_DataType TF_OperationOutputType(TF_Output oper_out);
607 TF_CAPI_EXPORT extern TF_Output TF_OperationInput(TF_Input oper_in);
612 TF_CAPI_EXPORT extern int TF_OperationOutputNumConsumers(TF_Output oper_out);
[all …]
Dc_test_util.h69 TF_Operation* Add(TF_Output l, TF_Output r, TF_Graph* graph, TF_Status* s,
83 TF_Operation* LessThan(TF_Output l, TF_Output r, TF_Graph* graph, TF_Status* s);
126 void SetOutputs(const std::vector<TF_Output>& outputs);
142 std::vector<TF_Output> inputs_;
144 std::vector<TF_Output> outputs_;
Dc_api_test.cc344 TF_Output feed_out_0 = TF_Output{feed, 0}; in TEST()
422 TF_Output three_out_0 = TF_Output{three, 0}; in TEST()
448 EXPECT_EQ(TF_INT32, TF_OperationOutputType(TF_Output{feed, 0})); in TEST()
452 EXPECT_EQ(0, TF_OperationOutputNumConsumers(TF_Output{feed, 0})); in TEST()
481 EXPECT_EQ(TF_INT32, TF_OperationOutputType(TF_Output{add, 0})); in TEST()
489 TF_Output add_in_0 = TF_OperationInput(TF_Input{add, 0}); in TEST()
492 TF_Output add_in_1 = TF_OperationInput(TF_Input{add, 1}); in TEST()
495 EXPECT_EQ(0, TF_OperationOutputNumConsumers(TF_Output{add, 0})); in TEST()
505 ASSERT_EQ(1, TF_OperationOutputNumConsumers(TF_Output{feed, 0})); in TEST()
507 EXPECT_EQ(1, TF_OperationOutputConsumers(TF_Output{feed, 0}, &feed_port, 1)); in TEST()
[all …]
Dc_test_util.cc133 TF_Output add_inputs[2] = {{l, 0}, {r, 0}}; in AddOpHelper()
160 TF_Output add_inputs[2] = {{l, 0}, {r, 0}}; in AddWithCtrlDependency()
196 TF_Operation* Add(TF_Output l, TF_Output r, TF_Graph* graph, TF_Status* s, in Add()
199 TF_Output inputs[2] = {l, r}; in Add()
207 TF_Output neg_input = {n, 0}; in NegHelper()
221 TF_Operation* LessThan(TF_Output l, TF_Output r, TF_Graph* graph, in LessThan()
426 inputs_.emplace_back(TF_Output{p.first, 0}); in SetInputs()
435 outputs_.emplace_back(TF_Output{o, 0}); in SetOutputs()
440 void CSession::SetOutputs(const std::vector<TF_Output>& outputs) { in SetOutputs()
461 const TF_Output* inputs_ptr = inputs_.empty() ? nullptr : &inputs_[0]; in Run()
[all …]
Dc_api.cc677 void TF_GraphSetOutputHandleShapesAndTypes(TF_Graph* graph, TF_Output output, in TF_GraphSetOutputHandleShapesAndTypes()
964 string OutputName(const TF_Output& output) { in OutputName()
979 TensorId ToTensorId(const TF_Output& output) { in ToTensorId()
984 std::vector<tensorflow::Output> OutputsFromTFOutputs(TF_Output* tf_outputs, in OutputsFromTFOutputs()
995 TF_Output* tf_outputs) { in TFOutputsFromOutputs()
1007 void TF_GraphSetTensorShape(TF_Graph* graph, TF_Output output, in TF_GraphSetTensorShape()
1025 int TF_GraphGetTensorNumDims(TF_Graph* graph, TF_Output output, in TF_GraphGetTensorNumDims()
1048 void TF_GraphGetTensorShape(TF_Graph* graph, TF_Output output, int64_t* dims, in TF_GraphGetTensorShape()
1112 void TF_AddInput(TF_OperationDescription* desc, TF_Output input) { in TF_AddInput()
1116 void TF_AddInputList(TF_OperationDescription* desc, const TF_Output* inputs, in TF_AddInputList()
[all …]
Dc_api_internal.h106 TF_Output* parent_inputs;
147 std::vector<TF_Output> return_tensors;
205 void TF_GraphSetOutputHandleShapesAndTypes(TF_Graph* graph, TF_Output output,
Dwhile_loop_test.cc79 void Run(const std::vector<TF_Output>& run_outputs, in Run()
128 std::vector<TF_Output> inputs_; // The inputs to the while loop
129 std::vector<TF_Output> outputs_; // The final outputs of the while loop
236 TF_Output inner_inputs[] = {params_->body_inputs[0], params_->body_inputs[1]}; in TEST_F()
267 TF_Output inner_outputs[2] = {{nullptr, -1}}; in TEST_F()
433 TF_Output grad_output; in TEST_F()
Dc_api_function_test.cc87 std::initializer_list<TF_Output> outputs, in RunT()
112 std::initializer_list<TF_Output> outputs, in Run()
143 std::vector<TF_Output> ToOutput(const std::vector<TF_Operation*> ops) { in ToOutput()
144 std::vector<TF_Output> out; in ToOutput()
176 const std::vector<TF_Output>& inputs, in DefineT()
177 const std::vector<TF_Output>& outputs, in DefineT()
203 TF_Operation* UseT(const std::vector<TF_Output>& inputs) { in UseT()
212 void UseHelper(const std::vector<TF_Output>& inputs, TF_Operation** op) { in UseHelper()
904 std::vector<TF_Output> outputs; in TEST_F()
909 std::vector<TF_Output> inputs = {{feed1, 0}, {feed2, 0}}; in TEST_F()
[all …]
Dpython_api.h35 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst,
Dc_api_function.cc389 const TF_Output* inputs, std::vector<OutputTensor>* input_tensors, in ProcessInputs()
425 int noutputs, const TF_Output* outputs, in ProcessOutputs()
490 int ninputs, const TF_Output* inputs, in TF_GraphToFunction()
491 int noutputs, const TF_Output* outputs, in TF_GraphToFunction()
Dpython_api.cc49 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst, in UpdateEdge()
/external/tensorflow/tensorflow/go/
Dsession.go203 cfeeds = make([]C.TF_Output, len(feeds))
204 cfetches = make([]C.TF_Output, len(fetches))
207 pcfeeds *C.TF_Output
208 pcfetches *C.TF_Output
342 feeds []C.TF_Output
344 fetches []C.TF_Output
351 fetches: make([]C.TF_Output, len(fetches)),
376 func ptrOutput(l []C.TF_Output) *C.TF_Output { argument
Doperation.go115 func (p Output) c() C.TF_Output {
121 return C.TF_Output{oper: p.Op.c, index: C.int(p.Index)}
Dgraph.go200 list := make([]C.TF_Output, size)
/external/tensorflow/tensorflow/java/src/main/native/
Dsession_jni.cc59 jintArray src_index, TF_Output* dst, jint n) { in resolveOutputs()
82 dst[i] = TF_Output{reinterpret_cast<TF_Operation*>(op_handles[i]), in resolveOutputs()
173 std::unique_ptr<TF_Output[]> inputs(new TF_Output[ninputs]); in Java_org_tensorflow_Session_run()
175 std::unique_ptr<TF_Output[]> outputs(new TF_Output[noutputs]); in Java_org_tensorflow_Session_run()
Doperation_jni.cc104 TF_Output output{op, output_index}; in Java_org_tensorflow_Operation_shape()
157 return static_cast<jint>(TF_OperationOutputType(TF_Output{op, output_index})); in Java_org_tensorflow_Operation_dtype()
Doperation_builder_jni.cc33 bool resolveOutput(JNIEnv* env, jlong op_handle, jint index, TF_Output* out) { in resolveOutput()
88 TF_Output out; in Java_org_tensorflow_OperationBuilder_addInput()
108 std::unique_ptr<TF_Output[]> o(new TF_Output[n]); in Java_org_tensorflow_OperationBuilder_addInputList()
/external/tensorflow/tensorflow/python/framework/
Dc_api_util.py109 ret = c_api.TF_Output()
/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc645 TF_Output input{arg, 0}; in TEST()
646 TF_Output output{id, 0}; in TEST()
707 TF_Output input{arg, 0}; in TEST()
708 TF_Output output{id, 0}; in TEST()
/external/tensorflow/tensorflow/contrib/tensorrt/segment/
Dsegment_test.cc90 TF_Output add_inputs[2] = {{l, 0}, {r, 0}}; in AddHelper()
/external/tensorflow/tensorflow/docs_src/extend/
Dlanguage_bindings.md86 method to fetch that output as a tensor. Corresponds to a `TF_Output` in the

12