Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.h142 TF_Graph* graph, TF_Output output, TF_Status* status, bool* unknown_shape);
150 const std::vector<TF_Output>& inputs,
152 const std::vector<TF_Output>& outputs,
167 const std::vector<TF_Output>& inputs,
168 const std::vector<TF_Output>& outputs,
183 const std::vector<TF_Output>& inputs,
185 const std::vector<TF_Output>& outputs,
190 std::vector<TF_Output> GetOperationInputs(TF_Operation* oper);
203 TF_Output oper_out);
210 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.cc358 const std::vector<TF_Output>& inputs, in TF_SessionRun_wrapper_helper()
360 const std::vector<TF_Output>& outputs, in TF_SessionRun_wrapper_helper()
444 const std::vector<TF_Output>& inputs, in TF_SessionRun_wrapper()
446 const std::vector<TF_Output>& outputs, in TF_SessionRun_wrapper()
494 TF_Graph* graph, TF_Output output, TF_Status* out_status, in TF_GraphGetTensorShapeHelper()
520 const std::vector<TF_Output>& inputs, in TF_SessionPRunSetup_wrapper()
521 const std::vector<TF_Output>& outputs, in TF_SessionPRunSetup_wrapper()
534 const std::vector<TF_Output>& inputs, in TF_SessionPRun_wrapper()
536 const std::vector<TF_Output>& outputs, in TF_SessionPRun_wrapper()
550 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()
209 %typemap(out) std::vector<TF_Output> tensorflow::GetOperationInputs {
216 const std::vector<TF_Output>& tf_outputs = $1;
380 %typemap(in) (const TF_Output* inputs, int num_inputs)
381 (std::vector<TF_Output> inputs) {
391 %typemap(in, numinputs=0) (int* num_outputs, TF_Output** outputs)
392 (int num_outputs, TF_Output* outputs) {
[all …]
/external/tensorflow/tensorflow/c/
Dc_api.h395 typedef struct TF_Output { struct
398 } TF_Output; argument
425 TF_Output output,
438 TF_Output output,
454 TF_Output output,
491 TF_Output input);
496 const TF_Output* inputs,
630 TF_CAPI_EXPORT extern TF_DataType TF_OperationOutputType(TF_Output oper_out);
645 TF_CAPI_EXPORT extern TF_Output TF_OperationInput(TF_Input oper_in);
650 TF_CAPI_EXPORT extern int TF_OperationOutputNumConsumers(TF_Output oper_out);
[all …]
Dc_test_util.h82 TF_Operation* Add(TF_Output l, TF_Output r, TF_Graph* graph, TF_Status* s,
99 TF_Operation* LessThan(TF_Output l, TF_Output r, TF_Graph* graph, TF_Status* s);
142 void SetOutputs(const std::vector<TF_Output>& outputs);
158 std::vector<TF_Output> inputs_;
160 std::vector<TF_Output> outputs_;
Dc_api_test.cc365 TF_Output feed_out_0 = TF_Output{feed, 0}; in TEST()
443 TF_Output three_out_0 = TF_Output{three, 0}; in TEST()
469 EXPECT_EQ(TF_INT32, TF_OperationOutputType(TF_Output{feed, 0})); in TEST()
473 EXPECT_EQ(0, TF_OperationOutputNumConsumers(TF_Output{feed, 0})); in TEST()
502 EXPECT_EQ(TF_INT32, TF_OperationOutputType(TF_Output{add, 0})); in TEST()
510 TF_Output add_in_0 = TF_OperationInput(TF_Input{add, 0}); in TEST()
513 TF_Output add_in_1 = TF_OperationInput(TF_Input{add, 1}); in TEST()
516 EXPECT_EQ(0, TF_OperationOutputNumConsumers(TF_Output{add, 0})); in TEST()
526 ASSERT_EQ(1, TF_OperationOutputNumConsumers(TF_Output{feed, 0})); in TEST()
528 EXPECT_EQ(1, TF_OperationOutputConsumers(TF_Output{feed, 0}, &feed_port, 1)); in TEST()
[all …]
Dpython_api.h43 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst,
65 std::string GetHandleShapeAndType(TF_Graph* graph, TF_Output output);
72 void SetHandleShapeAndType(TF_Graph* graph, TF_Output output, const void* proto,
78 void AddWhileInputHack(TF_Graph* graph, TF_Output new_src, TF_Operation* dst,
Dc_test_util.cc174 TF_Output add_inputs[2] = {{l, 0}, {r, 0}}; in AddOpHelper()
201 TF_Output add_inputs[2] = {{l, 0}, {r, 0}}; in AddWithCtrlDependency()
244 TF_Operation* Add(TF_Output l, TF_Output r, TF_Graph* graph, TF_Status* s, in Add()
247 TF_Output inputs[2] = {l, r}; in Add()
255 TF_Output neg_input = {n, 0}; in NegHelper()
269 TF_Operation* LessThan(TF_Output l, TF_Output r, TF_Graph* graph, in LessThan()
474 inputs_.emplace_back(TF_Output{p.first, 0}); in SetInputs()
483 outputs_.emplace_back(TF_Output{o, 0}); in SetOutputs()
488 void CSession::SetOutputs(const std::vector<TF_Output>& outputs) { in SetOutputs()
509 const TF_Output* inputs_ptr = inputs_.empty() ? nullptr : &inputs_[0]; in Run()
[all …]
Dc_api.cc721 void TF_GraphSetOutputHandleShapesAndTypes(TF_Graph* graph, TF_Output output, in TF_GraphSetOutputHandleShapesAndTypes()
1067 string OutputName(const TF_Output& output) { in OutputName()
1082 TensorId ToTensorId(const TF_Output& output) { in ToTensorId()
1087 std::vector<tensorflow::Output> OutputsFromTFOutputs(TF_Output* tf_outputs, in OutputsFromTFOutputs()
1098 TF_Output* tf_outputs) { in TFOutputsFromOutputs()
1110 void TF_GraphSetTensorShape(TF_Graph* graph, TF_Output output, in TF_GraphSetTensorShape()
1128 int TF_GraphGetTensorNumDims(TF_Graph* graph, TF_Output output, in TF_GraphGetTensorNumDims()
1151 void TF_GraphGetTensorShape(TF_Graph* graph, TF_Output output, int64_t* dims, in TF_GraphGetTensorShape()
1215 void TF_AddInput(TF_OperationDescription* desc, TF_Output input) { in TF_AddInput()
1219 void TF_AddInputList(TF_OperationDescription* desc, const TF_Output* inputs, in TF_AddInputList()
[all …]
Dc_api_internal.h111 TF_Output* parent_inputs;
153 std::vector<TF_Output> return_tensors;
221 void TF_GraphSetOutputHandleShapesAndTypes(TF_Graph* graph, TF_Output output,
235 std::string getTF_OutputDebugString(TF_Output node);
Dpython_api.cc59 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst, in UpdateEdge()
122 std::string GetHandleShapeAndType(TF_Graph* graph, TF_Output output) { in GetHandleShapeAndType()
147 void SetHandleShapeAndType(TF_Graph* graph, TF_Output output, const void* proto, in SetHandleShapeAndType()
172 void AddWhileInputHack(TF_Graph* graph, TF_Output new_src, TF_Operation* dst, in AddWhileInputHack()
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_experimental.h272 TF_Output t, TF_DataType data_type);
281 TF_CAPI_EXPORT extern TF_Output TFE_GetTFOutputFromTensorHandle(
309 TF_CAPI_EXPORT extern TF_Output TFE_GetInputGraphNodeFromTraceContext(
Dc_api_function_test.cc89 std::initializer_list<TF_Output> outputs, in RunT()
114 std::initializer_list<TF_Output> outputs, in Run()
145 std::vector<TF_Output> ToOutput(const std::vector<TF_Operation*> ops) { in ToOutput()
146 std::vector<TF_Output> out; in ToOutput()
178 const std::vector<TF_Output>& inputs, in DefineT()
179 const std::vector<TF_Output>& outputs, in DefineT()
206 TF_Operation* UseT(const std::vector<TF_Output>& inputs) { in UseT()
215 void UseHelper(const std::vector<TF_Output>& inputs, TF_Operation** op) { in UseHelper()
907 std::vector<TF_Output> outputs; in TEST_F()
912 std::vector<TF_Output> inputs = {{feed1, 0}, {feed2, 0}}; in TEST_F()
[all …]
Dc_api_function.cc462 const TF_Output* inputs, std::vector<OutputTensor>* input_tensors, in ProcessInputs()
498 int noutputs, const TF_Output* outputs, in ProcessOutputs()
563 const TF_Operation* const* opers, int ninputs, const TF_Output* inputs, in TF_GraphToFunctionWithControlOutputs()
564 int noutputs, const TF_Output* outputs, const char* const* output_names, in TF_GraphToFunctionWithControlOutputs()
630 int ninputs, const TF_Output* inputs, in TF_GraphToFunction()
631 int noutputs, const TF_Output* outputs, in TF_GraphToFunction()
/external/tensorflow/tensorflow/java/src/main/native/
Dgraph_jni.cc153 std::unique_ptr<TF_Output[]> y(new TF_Output[ny]); in Java_org_tensorflow_Graph_addGradients()
154 std::unique_ptr<TF_Output[]> x(new TF_Output[nx]); in Java_org_tensorflow_Graph_addGradients()
155 std::unique_ptr<TF_Output[]> dx(nullptr); in Java_org_tensorflow_Graph_addGradients()
156 std::unique_ptr<TF_Output[]> dy(new TF_Output[nx]); in Java_org_tensorflow_Graph_addGradients()
166 dx.reset(new TF_Output[ny]); in Java_org_tensorflow_Graph_addGradients()
191 TF_Output dy_output = dy.get()[i]; in Java_org_tensorflow_Graph_addGradients()
203 const TF_Output* const inputs, in buildSubgraph()
204 const TF_Output* const outputs, const int ninputs, in buildSubgraph()
266 std::unique_ptr<TF_Output[]> inputs(new TF_Output[ninputs]); in Java_org_tensorflow_Graph_whileLoop()
315 std::unique_ptr<TF_Output[]> outputs(new TF_Output[ninputs]); in Java_org_tensorflow_Graph_whileLoop()
[all …]
Dsession_jni.cc145 std::unique_ptr<TF_Output[]> inputs(new TF_Output[ninputs]); in Java_org_tensorflow_Session_run()
147 std::unique_ptr<TF_Output[]> outputs(new TF_Output[noutputs]); in Java_org_tensorflow_Session_run()
Dutils_jni.cc21 jintArray src_index, TF_Output* dst, jint n) { in resolveOutputs()
44 dst[i] = TF_Output{reinterpret_cast<TF_Operation*>(op_handles[i]), in resolveOutputs()
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()
Dutils_jni.h28 jintArray src_index, TF_Output* dst, jint n);
/external/tensorflow/tensorflow/go/
Dsession.go216 cfeeds = make([]C.TF_Output, len(feeds))
217 cfetches = make([]C.TF_Output, len(fetches))
220 pcfeeds *C.TF_Output
221 pcfetches *C.TF_Output
355 feeds []C.TF_Output
357 fetches []C.TF_Output
364 fetches: make([]C.TF_Output, len(fetches)),
389 func ptrOutput(l []C.TF_Output) *C.TF_Output { argument
Dgraph.go194 cy = make([]C.TF_Output, len(y))
195 cx = make([]C.TF_Output, len(x))
196 cdx = make([]C.TF_Output, len(dx))
197 cdy = make([]C.TF_Output, len(x))
199 pcy *C.TF_Output
200 pcx *C.TF_Output
201 pcdx *C.TF_Output
202 pcdy *C.TF_Output
304 list := make([]C.TF_Output, size)
Doperation.go126 func (p Output) c() C.TF_Output {
132 return C.TF_Output{oper: p.Op.c, index: C.int(p.Index)}
/external/tensorflow/tensorflow/c/eager/
Dc_api_internal.h88 TFE_TensorHandle(TF_Output t, TF_DataType dtype) in TFE_TensorHandle()
147 std::map<tensorflow::TensorHandle*, TF_Output> input_tensor_map;
148 std::vector<std::pair<tensorflow::TensorHandle*, TF_Output>>* input_tensors =

12