Home
last modified time | relevance | path

Searched refs:IsPlugin (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/
Dtrt_plugin_factory_test.cc89 if (PluginFactoryTensorRT::GetInstance()->IsPlugin( in RegisterStubPlugin()
100 PluginFactoryTensorRT::GetInstance()->IsPlugin(StubPlugin::kPluginName)); in TEST_F()
104 PluginFactoryTensorRT::GetInstance()->IsPlugin(StubPlugin::kPluginName)); in TEST_F()
110 PluginFactoryTensorRT::GetInstance()->IsPlugin(StubPlugin::kPluginName)); in TEST_F()
Dtrt_plugin_factory.cc32 if (!IsPlugin(encoded_op_name)) { in createPlugin()
45 if (!IsPlugin(op_name)) return nullptr; in CreatePlugin()
57 if (IsPlugin(op_name)) return false; in RegisterPlugin()
Dtrt_plugin_factory.h56 bool IsPlugin(const string& op_name) { in IsPlugin() function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc942 if (PluginFactoryTensorRT::GetInstance()->IsPlugin(op)) return Status::OK(); in ValidateNode()
1008 if (PluginFactoryTensorRT::GetInstance()->IsPlugin(op)) { in ConvertNode()