Searched refs:deserialize_func (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/ |
| D | trt_plugin_factory.h | 53 PluginDeserializeFunc deserialize_func,
77 TrtPluginRegistrar(const string& name, PluginDeserializeFunc deserialize_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
86 REGISTER_TRT_PLUGIN_UNIQ_HELPER(__COUNTER__, name, deserialize_func, \
88 #define REGISTER_TRT_PLUGIN_UNIQ_HELPER(ctr, name, deserialize_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 | 55 const string& op_name, PluginDeserializeFunc deserialize_func, in RegisterPlugin() argument
61 op_name, std::make_pair(deserialize_func, construct_func)); in RegisterPlugin()
|