Home
last modified time | relevance | path

Searched refs:TF_Input (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/go/
Doperation.go143 inputs := make([]C.TF_Input, max)
144 n := C.TF_OperationOutputConsumers(p.c(), (*C.TF_Input)(unsafe.Pointer(&inputs[0])), C.int(max))
172 func (p Consumer) c() C.TF_Input {
178 return C.TF_Input{oper: p.Op.c, index: C.int(p.Index)}
/external/tensorflow/tensorflow/c/
Dc_api.h189 typedef struct TF_Input { struct
192 } TF_Input; argument
436 TF_CAPI_EXPORT extern TF_DataType TF_OperationInputType(TF_Input oper_in);
445 TF_CAPI_EXPORT extern TF_Output TF_OperationInput(TF_Input oper_in);
469 TF_Input* consumers,
1529 TF_Input dst, TF_Status* status);
Dpython_api.h43 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst,
Dpython_api.cc58 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst, in UpdateEdge()
Dc_api_test.cc511 EXPECT_EQ(TF_INT32, TF_OperationInputType(TF_Input{add, 0})); in TEST()
512 EXPECT_EQ(TF_INT32, TF_OperationInputType(TF_Input{add, 1})); in TEST()
513 TF_Output add_in_0 = TF_OperationInput(TF_Input{add, 0}); in TEST()
516 TF_Output add_in_1 = TF_OperationInput(TF_Input{add, 1}); in TEST()
530 TF_Input feed_port; in TEST()
537 TF_Input three_port; in TEST()
661 TF_UpdateEdge(graph, TF_Output{one, 0}, TF_Input{neg, 0}, s); in TEST()
Dc_api.cc1119 TF_DataType TF_OperationInputType(TF_Input oper_in) { in TF_OperationInputType()
1137 TF_Output TF_OperationInput(TF_Input oper_in) { in TF_OperationInput()
1167 int TF_OperationOutputConsumers(TF_Output oper_out, TF_Input* consumers, in TF_OperationOutputConsumers()
2492 void TF_UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst, in TF_UpdateEdge()
/external/tensorflow/tensorflow/python/client/
Dtf_session_wrapper.cc110 py::class_<TF_Input>(m, "TF_Input") in PYBIND11_MODULE()
112 .def_readwrite("oper", &TF_Input::oper) in PYBIND11_MODULE()
113 .def_readwrite("index", &TF_Input::index); in PYBIND11_MODULE()
752 m.def("UpdateEdge", [](TF_Graph* graph, TF_Output new_src, TF_Input dst) { in PYBIND11_MODULE()
Dtf_session_helper.cc568 std::vector<TF_Input> consumers(num_consumers); in TF_OperationOutputConsumers_wrapper()
/external/tensorflow/tensorflow/python/framework/
Dops.py2225 tf_input = pywrap_tf_session.TF_Input()