File tree 1 file changed +7
-1
lines changed
paddle/fluid/inference/tensorrt
1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -47,6 +47,7 @@ struct SimpleOpTypeSetTeller : public Teller {
47
47
#endif
48
48
#if IS_TRT_VERSION_GE(7000)
49
49
teller_set.insert (" tile" );
50
+ int8_teller_set.insert (" tile" );
50
51
teller_set.insert (" flatten_contiguous_range" );
51
52
int8_teller_set.insert (" flatten_contiguous_range" );
52
53
teller_set.insert (" rnn" );
@@ -2302,15 +2303,20 @@ struct SimpleOpTypeSetTeller : public Teller {
2302
2303
if (!with_dynamic_shape) {
2303
2304
if (tile_inputs.find (" repeat_times_tensor" ) != tile_inputs.end ()) {
2304
2305
if (!desc.Input (" repeat_times_tensor" ).empty ()) {
2306
+ VLOG (3 ) << " Tile op: repeat_times_tensor is not empty." ;
2305
2307
return false ;
2306
2308
}
2307
2309
}
2308
2310
if (tile_inputs.find (" RepeatTimes" ) != tile_inputs.end ()) {
2309
2311
if (!desc.Input (" RepeatTimes" ).empty ()) {
2312
+ VLOG (3 ) << " Tile op: RepeatTimes is not empty." ;
2310
2313
return false ;
2311
2314
}
2312
2315
}
2313
- if (!desc.HasAttr (" repeat_times" )) return false ;
2316
+ if (!desc.HasAttr (" repeat_times" )) {
2317
+ VLOG (3 ) << " Tile op:`repeat_times` is not set." ;
2318
+ return false ;
2319
+ }
2314
2320
}
2315
2321
}
2316
2322
#endif
You can’t perform that action at this time.
0 commit comments