Searched refs:serial_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/ |
| D | trt_plugin_utils.cc | 25 string ExtractOpName(const void* serial_data, size_t serial_length, in ExtractOpName() argument
27 size_t op_name_char_count = *static_cast<const size_t*>(serial_data); in ExtractOpName()
32 const char* buffer = static_cast<const char*>(serial_data) + sizeof(size_t); in ExtractOpName()
|
| D | trt_plugin_factory.cc | 25 const void* serial_data, in createPlugin() argument
30 ExtractOpName(serial_data, serial_length, &parsed_byte); in createPlugin()
38 plugin_registry_[encoded_op_name].first(serial_data, serial_length); in createPlugin()
|
| D | trt_plugin_utils.h | 37 string ExtractOpName(const void* serial_data, size_t serial_length,
|
| D | trt_plugin_factory.h | 46 PluginTensorRT* createPlugin(const char* layer_name, const void* serial_data,
|