Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/
Dc_api.h381 typedef struct TF_OperationDescription TF_OperationDescription; typedef
462 TF_CAPI_EXPORT extern TF_OperationDescription* TF_NewOperation(
466 TF_CAPI_EXPORT extern void TF_SetDevice(TF_OperationDescription* desc,
490 TF_CAPI_EXPORT extern void TF_AddInput(TF_OperationDescription* desc,
495 TF_CAPI_EXPORT extern void TF_AddInputList(TF_OperationDescription* desc,
500 TF_CAPI_EXPORT extern void TF_AddControlInput(TF_OperationDescription* desc,
508 TF_CAPI_EXPORT extern void TF_ColocateWith(TF_OperationDescription* desc,
516 TF_CAPI_EXPORT extern void TF_SetAttrString(TF_OperationDescription* desc,
521 TF_CAPI_EXPORT extern void TF_SetAttrStringList(TF_OperationDescription* desc,
526 TF_CAPI_EXPORT extern void TF_SetAttrInt(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.cc1198 static TF_OperationDescription* TF_NewOperationLocked(TF_Graph* graph, in TF_NewOperationLocked()
1202 return new TF_OperationDescription(graph, op_type, oper_name); in TF_NewOperationLocked()
1205 TF_OperationDescription* TF_NewOperation(TF_Graph* graph, const char* op_type, in TF_NewOperation()
1211 void TF_SetDevice(TF_OperationDescription* desc, const char* device) { in TF_SetDevice()
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()
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()
1238 void TF_SetAttrString(TF_OperationDescription* desc, const char* attr_name, in TF_SetAttrString()
1244 void TF_SetAttrStringList(TF_OperationDescription* desc, const char* attr_name, in TF_SetAttrStringList()
[all …]
Dc_api_internal.h114 struct TF_OperationDescription { struct
115 TF_OperationDescription(TF_Graph* g, const char* op_type, in TF_OperationDescription() argument
Dc_api_test.cc1218 void SetViaStringList(TF_OperationDescription* desc, in SetViaStringList()
1227 void SetViaProto(TF_OperationDescription* desc, in SetViaProto()
1267 void FinishAndVerify(TF_OperationDescription* desc, in FinishAndVerify()
1279 TF_OperationDescription* desc_;
1720 TF_OperationDescription* desc = TF_NewOperation(graph, "Const", name); in FloatConst2x2()
1732 TF_OperationDescription* desc = TF_NewOperation(graph, "MatMul", name); in MatMul()
1748 TF_OperationDescription* desc = TF_NewOperation(graph, "OnesLike", name); in OnesLike()
1757 TF_OperationDescription* desc = in NoGradientOp()
1933 TF_OperationDescription* init(string type) { in init()
Dwhile_loop_test.cc399 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()
1689 TF_OperationDescription* desc = in TEST_F()
/external/tensorflow/tensorflow/java/src/main/native/
Doperation_builder_jni.cc24 TF_OperationDescription* requireHandle(JNIEnv* env, jlong handle) { in requireHandle()
30 return reinterpret_cast<TF_OperationDescription*>(handle); in requireHandle()
64 TF_OperationDescription* d = TF_NewOperation(graph, op_type, op_name); in Java_org_tensorflow_OperationBuilder_allocate()
67 static_assert(sizeof(jlong) >= sizeof(TF_OperationDescription*), in Java_org_tensorflow_OperationBuilder_allocate()
74 TF_OperationDescription* d = requireHandle(env, handle); in Java_org_tensorflow_OperationBuilder_finish()
90 TF_OperationDescription* d = requireHandle(env, handle); in Java_org_tensorflow_OperationBuilder_addInput()
98 TF_OperationDescription* d = requireHandle(env, handle); in Java_org_tensorflow_OperationBuilder_addInputList()
130 TF_OperationDescription* d = requireHandle(env, handle); in Java_org_tensorflow_OperationBuilder_addControlInput()
137 TF_OperationDescription* d = requireHandle(env, handle); in Java_org_tensorflow_OperationBuilder_setDevice()
148 TF_OperationDescription* d = requireHandle(env, handle); in Java_org_tensorflow_OperationBuilder_setAttrString()
[all …]
/external/tensorflow/tensorflow/go/
Dgraph.go344 func setAttr(cdesc *C.TF_OperationDescription, status *status, name string, value interface{}) erro… argument
/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc1170 TF_OperationDescription* arg_descr = in TEST()
1176 TF_OperationDescription* id_descr = in TEST()
1237 TF_OperationDescription* arg_descr = in TEST()
1243 TF_OperationDescription* id_descr = in TEST()