@@ -272,6 +272,11 @@ public void init(ClassProperties properties) {
272
272
@ Override
273
273
public void map (InfoMap infoMap ) {
274
274
infoMap
275
+ .put (
276
+ new Info ("c_api_experimental.h" )
277
+ .linePatterns (
278
+ "typedef struct TFE_OpAttrs TFE_OpAttrs;" ,
279
+ "#define TFE_CUSTOM_DEVICE_VERSION 4" ))
275
280
.put (new Info ("TF_CAPI_EXPORT" , "TF_Bool" ).cppTypes ().annotations ())
276
281
.put (
277
282
new Info ("TF_Buffer::data" )
@@ -315,49 +320,10 @@ public void map(InfoMap infoMap) {
315
320
new Info ("TF_ImportGraphDefOptions" )
316
321
.pointerTypes ("TF_ImportGraphDefOptions" )
317
322
.base ("org.tensorflow.internal.c_api.AbstractTF_ImportGraphDefOptions" ))
318
- .put (
319
- new Info (
320
- "TF_Operation" ,
321
- "TF_WhileParams" ,
322
- "TFE_MonitoringCounterCell" ,
323
- "TFE_MonitoringSamplerCell" ,
324
- "TFE_MonitoringCounter0" ,
325
- "TFE_MonitoringCounter1" ,
326
- "TFE_MonitoringCounter2" ,
327
- "TFE_MonitoringIntGaugeCell" ,
328
- "TFE_MonitoringStringGaugeCell" ,
329
- "TFE_MonitoringBoolGaugeCell" ,
330
- "TFE_MonitoringIntGauge0" ,
331
- "TFE_MonitoringIntGauge1" ,
332
- "TFE_MonitoringIntGauge2" ,
333
- "TFE_MonitoringStringGauge0" ,
334
- "TFE_MonitoringStringGauge1" ,
335
- "TFE_MonitoringStringGauge2" ,
336
- "TFE_MonitoringBoolGauge0" ,
337
- "TFE_MonitoringBoolGauge1" ,
338
- "TFE_MonitoringBoolGauge2" ,
339
- "TFE_MonitoringSampler0" ,
340
- "TFE_MonitoringSampler1" ,
341
- "TFE_MonitoringSampler2" )
342
- .purify ())
323
+ .put (new Info ("TF_Operation" , "TF_WhileParams" ).purify ())
343
324
.put (
344
325
new Info ("TF_Operation::node" )
345
326
.javaText ("public native @MemberGetter @ByRef Node node();" ))
346
- .put (
347
- new Info ("TFE_MonitoringCounterCell::cell" )
348
- .javaText ("public native @MemberGetter @ByRef CounterCell cell();" ))
349
- .put (
350
- new Info ("TFE_MonitoringSamplerCell::cell" )
351
- .javaText ("public native @MemberGetter @ByRef SamplerCell cell();" ))
352
- .put (
353
- new Info ("TFE_MonitoringIntGaugeCell::cell" )
354
- .javaText ("public native @MemberGetter @ByRef IntGaugeCell cell();" ))
355
- .put (
356
- new Info ("TFE_MonitoringStringGaugeCell::cell" )
357
- .javaText ("public native @MemberGetter @ByRef StringGaugeCell cell();" ))
358
- .put (
359
- new Info ("TFE_MonitoringBoolGaugeCell::cell" )
360
- .javaText ("public native @MemberGetter @ByRef BoolGaugeCell cell();" ))
361
327
.put (
362
328
new Info ("TFE_Context" )
363
329
.pointerTypes ("TFE_Context" )
@@ -385,8 +351,29 @@ public void map(InfoMap infoMap) {
385
351
new Info (
386
352
"TF_ShapeInferenceContextDimValueKnown" ,
387
353
"TFE_NewTensorHandle(const tensorflow::Tensor&, TF_Status*)" ,
388
- "TF_InitKernel" )
389
- .skip ())
390
- .put (new Info ("TFE_CustomDeviceTensorHandle" , "TFE_CustomDevice" ).skip ());
354
+ "TF_InitKernel" ,
355
+ "TFE_MonitoringCounterCell" ,
356
+ "TFE_MonitoringCounter0" ,
357
+ "TFE_MonitoringCounter1" ,
358
+ "TFE_MonitoringCounter2" ,
359
+ "TFE_MonitoringIntGaugeCell" ,
360
+ "TFE_MonitoringIntGauge0" ,
361
+ "TFE_MonitoringIntGauge1" ,
362
+ "TFE_MonitoringIntGauge2" ,
363
+ "TFE_MonitoringStringGaugeCell" ,
364
+ "TFE_MonitoringStringGauge0" ,
365
+ "TFE_MonitoringStringGauge1" ,
366
+ "TFE_MonitoringStringGauge2" ,
367
+ "TFE_MonitoringBoolGaugeCell" ,
368
+ "TFE_MonitoringBoolGauge0" ,
369
+ "TFE_MonitoringBoolGauge1" ,
370
+ "TFE_MonitoringBoolGauge2" ,
371
+ "TFE_MonitoringSamplerCell" ,
372
+ "TFE_MonitoringSampler0" ,
373
+ "TFE_MonitoringSampler1" ,
374
+ "TFE_MonitoringSampler2" ,
375
+ "TFE_CustomDeviceTensorHandle" ,
376
+ "TFE_CustomDevice" )
377
+ .skip ());
391
378
}
392
379
}
0 commit comments