Searched refs:construct_func (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/ |
| D | trt_plugin_factory.h | 54 PluginConstructFunc construct_func);
78 PluginConstructFunc construct_func) { in TrtPluginRegistrar() argument
80 QCHECK(factory->RegisterPlugin(name, deserialize_func, construct_func)) in TrtPluginRegistrar()
85 #define REGISTER_TRT_PLUGIN(name, deserialize_func, construct_func) \ argument
87 construct_func)
89 construct_func) \ argument
90 REGISTER_TRT_PLUGIN_UNIQ(ctr, name, deserialize_func, construct_func)
91 #define REGISTER_TRT_PLUGIN_UNIQ(ctr, name, deserialize_func, construct_func) \ argument
93 TrtPluginRegistrar(name, deserialize_func, construct_func)
|
| D | trt_plugin_factory.cc | 56 PluginConstructFunc construct_func) { in RegisterPlugin() argument
61 op_name, std::make_pair(deserialize_func, construct_func)); in RegisterPlugin()
|