Searched refs:PluginTensorRT (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/ |
| D | inc_op_plugin.h | 31 class IncOpPlugin : public PluginTensorRT {
61 PluginTensorRT::configure(inputs, num_inputs, outputs, num_outputs, in configure()
75 return PluginTensorRT::getSerializationSize() + sizeof(float); in getSerializationSize()
80 PluginTensorRT::serialize(buffer); in serialize()
83 static_cast<char*>(buffer) + PluginTensorRT::getSerializationSize(); in serialize()
|
| D | inc_op_plugin.cc | 40 : PluginTensorRT(serialized_data, length), plugin_name_(kPluginName) { in IncOpPlugin()
42 size_t consumed_data = PluginTensorRT::getSerializationSize(); in IncOpPlugin()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/ |
| D | trt_plugin.cc | 29 PluginTensorRT::PluginTensorRT(const void* serialized_data, size_t length) { in PluginTensorRT() function in tensorflow::tensorrt::PluginTensorRT
50 void PluginTensorRT::configure(const nvinfer1::Dims* inputs, int num_inputs, in configure()
64 size_t PluginTensorRT::getSerializationSize() { in getSerializationSize()
71 void PluginTensorRT::serialize(void* serialized_data) { in serialize()
95 bool PluginTensorRT::StoreAttribute(const string& key, const void* ptr, in StoreAttribute()
|
| D | trt_plugin.h | 37 class PluginTensorRT : public nvinfer1::IPlugin {
39 PluginTensorRT() {} in PluginTensorRT() function
40 PluginTensorRT(const void* serialized_data, size_t length);
|
| D | trt_plugin_factory.h | 46 PluginTensorRT* createPlugin(const char* layer_name, const void* serial_data,
50 PluginTensorRT* CreatePlugin(const string& op_name);
71 std::vector<std::unique_ptr<PluginTensorRT>> owned_plugins_;
|
| D | trt_plugin_utils.h | 31 typedef std::function<PluginTensorRT*(const void*, size_t)>
34 typedef std::function<PluginTensorRT*(void)> PluginConstructFunc;
|
| D | trt_plugin_factory.cc | 24 PluginTensorRT* PluginFactoryTensorRT::createPlugin(const char* layer_name, in createPlugin()
44 PluginTensorRT* PluginFactoryTensorRT::CreatePlugin(const string& op_name) { in CreatePlugin()
|
| D | trt_plugin_factory_test.cc | 32 class StubPlugin : public PluginTensorRT {
39 : PluginTensorRT(serialized_data, length) {} in StubPlugin()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2016 PluginTensorRT* plugin = in ConvertPlugin()
|