Home
last modified time | relevance | path

Searched refs:TrtUniquePtrType (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles_test.cc73 builder_ = TrtUniquePtrType<nvinfer1::IBuilder>( in SetUp()
76 network_ = TrtUniquePtrType<nvinfer1::INetworkDefinition>( in SetUp()
78 builder_config_ = TrtUniquePtrType<nvinfer1::IBuilderConfig>( in SetUp()
82 network_ = TrtUniquePtrType<nvinfer1::INetworkDefinition>( in SetUp()
109 TrtUniquePtrType<nvinfer1::IBuilder> builder_;
110 TrtUniquePtrType<nvinfer1::INetworkDefinition> network_;
112 TrtUniquePtrType<nvinfer1::IBuilderConfig> builder_config_;
114 TrtUniquePtrType<nvinfer1::ICudaEngine> engine;
137 engine = TrtUniquePtrType<nvinfer1::ICudaEngine>( in TEST_F()
140 engine = TrtUniquePtrType<nvinfer1::ICudaEngine>( in TEST_F()
[all …]
Dtrt_lru_cache.h123 EngineContext(TrtUniquePtrType<nvinfer1::ICudaEngine>&& input_cuda_engine, in EngineContext()
128 EngineContext(TrtUniquePtrType<nvinfer1::ICudaEngine>&& input_cuda_engine, in EngineContext()
134 TrtUniquePtrType<nvinfer1::ICudaEngine> cuda_engine;
177 TrtUniquePtrType<nvinfer1::IBuilder> builder_;
178 TrtUniquePtrType<nvinfer1::ICudaEngine> engine_;
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_resource_ops_test.cc70 TrtUniquePtrType<nvinfer1::ICudaEngine> CreateTRTEngine() { in CreateTRTEngine()
71 TrtUniquePtrType<nvinfer1::IBuilder> builder( in CreateTRTEngine()
73 TrtUniquePtrType<nvinfer1::INetworkDefinition> network( in CreateTRTEngine()
97 TrtUniquePtrType<nvinfer1::ICudaEngine> engine( in CreateTRTEngine()
153 TrtUniquePtrType<nvinfer1::ICudaEngine> engine = CreateTRTEngine(); in TEST_F()
Dtrt_engine_resource_ops.cc134 TrtUniquePtrType<IRuntime> infer( in Compute()
137 TrtUniquePtrType<nvinfer1::ICudaEngine> engine( in Compute()
219 TrtUniquePtrType<nvinfer1::IHostMemory> engine_data( in Compute()
Dtrt_engine_op.cc144 StatusOr<TrtUniquePtrType<nvinfer1::ICudaEngine>> BuildEngine(
827 StatusOr<TrtUniquePtrType<nvinfer1::ICudaEngine>> TRTEngineOp::BuildEngine( in BuildEngine()
842 TrtUniquePtrType<nvinfer1::ICudaEngine> engine; in BuildEngine()
893 TrtUniquePtrType<IRuntime> infer(nvinfer1::createInferRuntime(logger)); in GetEngine()
898 TrtUniquePtrType<nvinfer1::ICudaEngine> static_engine( in GetEngine()
997 TrtUniquePtrType<nvinfer1::ICudaEngine> engine = in GetEngine()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h153 TrtUniquePtrType<nvinfer1::ICudaEngine>* engine, bool use_calibration,
479 Status BuildCudaEngine(TrtUniquePtrType<nvinfer1::ICudaEngine>* engine,
641 TrtUniquePtrType<nvinfer1::IBuilder> trt_builder_;
644 TrtUniquePtrType<nvinfer1::INetworkDefinition> trt_network_;
Dutils.h53 using TrtUniquePtrType = std::unique_ptr<T, TrtDestroyer<T>>; variable
Dconvert_graph.cc451 TrtUniquePtrType<nvinfer1::ICudaEngine> engine; in CreateTRTNode()
460 TrtUniquePtrType<nvinfer1::IHostMemory> engine_data(engine->serialize()); in CreateTRTNode()
Dconvert_nodes.cc1450 TrtUniquePtrType<nvinfer1::ICudaEngine>* engine, int max_batch_size, in BuildCudaEngine()
1465 TrtUniquePtrType<nvinfer1::IBuilderConfig> builder_config( in BuildCudaEngine()
6049 TrtUniquePtrType<nvinfer1::IPluginV2> plugin( in ConvertCombinedNMS()
6325 TrtUniquePtrType<nvinfer1::ICudaEngine>* engine, bool use_calibration, in ConvertGraphDefToEngine()
Dconvert_nodes_test.cc1298 TrtUniquePtrType<nvinfer1::ICudaEngine> engine_;
1510 TrtUniquePtrType<nvinfer1::IExecutionContext> execution_context( in BuildAndRun()
1711 TrtUniquePtrType<nvinfer1::ICudaEngine> engine_;