Searched refs:TrtUniquePtrType (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
D | trt_shape_optimization_profiles_test.cc | 73 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 …]
|
D | trt_lru_cache.h | 123 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/ |
D | trt_engine_resource_ops_test.cc | 70 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()
|
D | trt_engine_resource_ops.cc | 134 TrtUniquePtrType<IRuntime> infer( in Compute() 137 TrtUniquePtrType<nvinfer1::ICudaEngine> engine( in Compute() 219 TrtUniquePtrType<nvinfer1::IHostMemory> engine_data( in Compute()
|
D | trt_engine_op.cc | 144 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/ |
D | convert_nodes.h | 153 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_;
|
D | utils.h | 53 using TrtUniquePtrType = std::unique_ptr<T, TrtDestroyer<T>>; variable
|
D | convert_graph.cc | 451 TrtUniquePtrType<nvinfer1::ICudaEngine> engine; in CreateTRTNode() 460 TrtUniquePtrType<nvinfer1::IHostMemory> engine_data(engine->serialize()); in CreateTRTNode()
|
D | convert_nodes.cc | 1450 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()
|
D | convert_nodes_test.cc | 1298 TrtUniquePtrType<nvinfer1::ICudaEngine> engine_; 1510 TrtUniquePtrType<nvinfer1::IExecutionContext> execution_context( in BuildAndRun() 1711 TrtUniquePtrType<nvinfer1::ICudaEngine> engine_;
|