Home
last modified time | relevance | path

Searched refs:TF_OperationDescription (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/c/
Dc_api.h164 typedef struct TF_OperationDescription TF_OperationDescription; typedef
247 TF_CAPI_EXPORT extern TF_OperationDescription* TF_NewOperationLocked(
254 TF_CAPI_EXPORT extern TF_OperationDescription* TF_NewOperation(
258 TF_CAPI_EXPORT extern void TF_SetDevice(TF_OperationDescription* desc,
282 TF_CAPI_EXPORT extern void TF_AddInput(TF_OperationDescription* desc,
287 TF_CAPI_EXPORT extern void TF_AddInputList(TF_OperationDescription* desc,
292 TF_CAPI_EXPORT extern void TF_AddControlInput(TF_OperationDescription* desc,
300 TF_CAPI_EXPORT extern void TF_ColocateWith(TF_OperationDescription* desc,
308 TF_CAPI_EXPORT extern void TF_SetAttrString(TF_OperationDescription* desc,
313 TF_CAPI_EXPORT extern void TF_SetAttrStringList(TF_OperationDescription* desc,
[all …]
Dc_test_util.cc111 TF_OperationDescription* desc = TF_NewOperation(graph, "Placeholder", name); in PlaceholderHelper()
130 TF_OperationDescription* desc = TF_NewOperation(graph, "Const", name); in ConstHelper()
173 TF_OperationDescription* desc = TF_NewOperation(graph, "AddN", name); in AddOpHelper()
200 TF_OperationDescription* desc = TF_NewOperation(graph, "AddN", name); in AddWithCtrlDependency()
211 TF_OperationDescription* desc = TF_NewOperation(graph, op_name, name); in BinaryOpHelper()
246 TF_OperationDescription* desc = TF_NewOperation(graph, "AddN", name); in Add()
254 TF_OperationDescription* desc = TF_NewOperation(graph, "Neg", name); in NegHelper()
271 TF_OperationDescription* desc = TF_NewOperation(graph, "Less", "less_than"); in LessThan()
279 TF_OperationDescription* desc = in RandomUniform()
290 TF_OperationDescription* desc = TF_NewOperation(graph, "Split", name); in Split3Helper()
Dc_api.cc776 TF_OperationDescription* TF_NewOperationLocked(TF_Graph* graph, in TF_NewOperationLocked()
780 return new TF_OperationDescription(graph, op_type, oper_name); in TF_NewOperationLocked()
783 TF_OperationDescription* TF_NewOperation(TF_Graph* graph, const char* op_type, in TF_NewOperation()
789 void TF_SetDevice(TF_OperationDescription* desc, const char* device) { in TF_SetDevice()
793 void TF_AddInput(TF_OperationDescription* desc, TF_Output input) { in TF_AddInput()
797 void TF_AddInputList(TF_OperationDescription* desc, const TF_Output* inputs, in TF_AddInputList()
807 void TF_AddControlInput(TF_OperationDescription* desc, TF_Operation* input) { in TF_AddControlInput()
811 void TF_ColocateWith(TF_OperationDescription* desc, TF_Operation* op) { in TF_ColocateWith()
816 void TF_SetAttrString(TF_OperationDescription* desc, const char* attr_name, in TF_SetAttrString()
822 void TF_SetAttrStringList(TF_OperationDescription* desc, const char* attr_name, in TF_SetAttrStringList()
[all …]
Dc_api_internal.h106 struct TF_OperationDescription { struct
107 TF_OperationDescription(TF_Graph* g, const char* op_type, in TF_OperationDescription() function
Dc_api_test.cc1258 void SetViaStringList(TF_OperationDescription* desc, in SetViaStringList()
1267 void SetViaProto(TF_OperationDescription* desc, in SetViaProto()
1307 void FinishAndVerify(TF_OperationDescription* desc, in FinishAndVerify()
1319 TF_OperationDescription* desc_;
1789 TF_OperationDescription* desc = TF_NewOperation(graph, "Const", name); in FloatConst2x2()
1801 TF_OperationDescription* desc = TF_NewOperation(graph, "MatMul", name); in MatMul()
1817 TF_OperationDescription* desc = TF_NewOperation(graph, "OnesLike", name); in OnesLike()
1826 TF_OperationDescription* desc = in NoGradientOp()
2002 TF_OperationDescription* init(string type) { in init()
Dwhile_loop_test.cc398 TF_OperationDescription* desc = TF_NewOperation( in TEST_F()
Dc_api_function_test.cc216 TF_OperationDescription* desc = in UseHelper()
1242 TF_OperationDescription* desc = TF_NewOperation(graph, "CustomOp", name); in NodeWithPlaceholderAttrHelper()
1283 TF_OperationDescription* desc = TF_NewOperation(graph, "Placeholder", name); in NodeWithAttrHelper()
/external/tensorflow/tensorflow/java/src/main/native/
Dgraph_operation_builder_jni.cc23 TF_OperationDescription* requireHandle(JNIEnv* env, jlong handle) { in requireHandle()
29 return reinterpret_cast<TF_OperationDescription*>(handle); in requireHandle()
63 TF_OperationDescription* d = TF_NewOperation(graph, op_type, op_name); in Java_org_tensorflow_GraphOperationBuilder_allocate()
66 static_assert(sizeof(jlong) >= sizeof(TF_OperationDescription*), in Java_org_tensorflow_GraphOperationBuilder_allocate()
73 TF_OperationDescription* d = requireHandle(env, handle); in Java_org_tensorflow_GraphOperationBuilder_finish()
89 TF_OperationDescription* d = requireHandle(env, handle); in Java_org_tensorflow_GraphOperationBuilder_addInput()
97 TF_OperationDescription* d = requireHandle(env, handle); in Java_org_tensorflow_GraphOperationBuilder_addInputList()
132 TF_OperationDescription* d = requireHandle(env, handle); in Java_org_tensorflow_GraphOperationBuilder_addControlInput()
139 TF_OperationDescription* d = requireHandle(env, handle); in Java_org_tensorflow_GraphOperationBuilder_setDevice()
150 TF_OperationDescription* d = requireHandle(env, handle); in Java_org_tensorflow_GraphOperationBuilder_setAttrString()
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
Dc_api_session_coordination_test.cc100 TF_OperationDescription* set_desc = in TEST_P()
108 TF_OperationDescription* get_desc = in TEST_P()
/external/tensorflow/tensorflow/python/client/
Dtf_session_wrapper.cc97 PYBIND11_MAKE_OPAQUE(TF_OperationDescription);
129 py::class_<TF_OperationDescription> TF_OperationDescription_class( in PYBIND11_MODULE()
587 TF_OperationDescription* output = in PYBIND11_MODULE()
596 [](TF_OperationDescription* desc) { in PYBIND11_MODULE()
623 m.def("TF_SetAttrValueProto", [](TF_OperationDescription* desc, in PYBIND11_MODULE()
768 "TF_AddInputList", [](TF_OperationDescription* desc, py::handle& inputs) { in PYBIND11_MODULE()
Dtf_session_helper.cc638 TF_OperationDescription* desc = in CreatePlaceholder()
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental_graph.cc122 op_.reset(new TF_OperationDescription(g_, op_type_.c_str(), in SetOpName()
306 std::unique_ptr<TF_OperationDescription> op_;
Dc_api_experimental_test.cc248 TF_OperationDescription* arg_descr = in TEST()
254 TF_OperationDescription* id_descr = in TEST()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_test.cc466 TF_OperationDescription* placeholder_desc = in RegisterCollectiveMulFunction()
473 TF_OperationDescription* reduce_desc = in RegisterCollectiveMulFunction()
/external/tensorflow/tensorflow/go/
Dgraph.go367 func setAttr(cdesc *C.TF_OperationDescription, status *status, name string, value interface{}) erro… argument