Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_lru_cache.h131 TrtUniquePtrType<nvinfer1::ICudaEngine>&& input_cuda_engine, in EngineContext()
132 TrtUniquePtrType<nvinfer1::IExecutionContext>&& input_execution_context) in EngineContext()
137 TrtUniquePtrType<nvinfer1::ICudaEngine> cuda_engine;
138 TrtUniquePtrType<nvinfer1::IExecutionContext> execution_context
Dtrt_resources.h60 TrtUniquePtrType<nvinfer1::IBuilder> builder_;
61 TrtUniquePtrType<nvinfer1::ICudaEngine> engine_;
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc557 TrtUniquePtrType<IRuntime> infer(nvinfer1::createInferRuntime(logger)); in GetEngine()
559 TrtUniquePtrType<nvinfer1::ICudaEngine> static_engine( in GetEngine()
577 TrtUniquePtrType<nvinfer1::IExecutionContext>( in GetEngine()
611 TrtUniquePtrType<nvinfer1::ICudaEngine> engine; in GetEngine()
637 TrtUniquePtrType<nvinfer1::IExecutionContext> exec_context( in GetEngine()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.h34 using TrtUniquePtrType = std::unique_ptr<T, TrtDestroyer<T>>; variable
Dconvert_nodes.h154 TrtUniquePtrType<nvinfer1::ICudaEngine>* engine, bool use_calibration,
Dconvert_nodes_test.cc571 TrtUniquePtrType<nvinfer1::IBuilder> builder_;
572 TrtUniquePtrType<nvinfer1::INetworkDefinition> network_;
1019 TrtUniquePtrType<nvinfer1::ICudaEngine> engine_;
1167 TrtUniquePtrType<nvinfer1::IExecutionContext> execution_context( in BuildAndRun()
1291 TrtUniquePtrType<nvinfer1::IBuilder> builder_;
1292 TrtUniquePtrType<nvinfer1::INetworkDefinition> network_;
1293 TrtUniquePtrType<nvinfer1::ICudaEngine> engine_;
Dconvert_graph.cc513 TrtUniquePtrType<nvinfer1::ICudaEngine> engine; in CreateTRTNode()
522 TrtUniquePtrType<nvinfer1::IHostMemory> engine_data(engine->serialize()); in CreateTRTNode()
Dconvert_nodes.cc4060 TrtUniquePtrType<nvinfer1::ICudaEngine>* engine, bool use_calibration, in ConvertGraphDefToEngine()
4066 TrtUniquePtrType<nvinfer1::IBuilder> builder( in ConvertGraphDefToEngine()
4088 TrtUniquePtrType<nvinfer1::INetworkDefinition>(builder->createNetwork()); in ConvertGraphDefToEngine()