Home
last modified time | relevance | path

Searched refs:StoreAttribute (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/
Dtrt_plugin.h55 virtual bool StoreAttribute(const string& key, const void* ptr,
Dtrt_plugin.cc95 bool PluginTensorRT::StoreAttribute(const string& key, const void* ptr, in StoreAttribute() function in tensorflow::tensorrt::PluginTensorRT
/external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/
Dinc_op_plugin.cc51 StoreAttribute(key, ptr, size); // save the attribute to own the data; in SetAttribute()