Searched refs:EngineContext (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 97 LRUCache<std::vector<TensorShape>, std::unique_ptr<EngineContext>,
121 Status ExecuteTrtEngine(OpKernelContext* ctx, EngineContext* engine_context,
137 StatusOr<std::pair<EngineContext*, int>> GetEngine(
697 StatusOr<std::pair<EngineContext*, int>> status = in ComputeAsync()
701 EngineContext* engine_context = status.ValueOrDie().first; in ComputeAsync()
742 EngineContext* engine_context, in ExecuteTrtEngine()
856 absl::make_unique<EngineContext>()); in BuildEngine()
862 StatusOr<std::pair<EngineContext*, int>> TRTEngineOp::GetEngine( in GetEngine()
865 static EngineContext empty_context; in GetEngine()
877 return std::pair<EngineContext*, int>(&empty_context, 0); in GetEngine()
[all …]
|
| D | trt_engine_op_test.cc | 225 EngineContext* ectx = cache->at({input_shape}).get(); in TEST_F()
251 EngineContext* ectx = cache->at({input_shape}).get(); in TEST_F()
279 EngineContext* ectx = cache->at({input_shape}).get(); in TEST_F()
|
| D | trt_engine_resource_ops.cc | 158 absl::make_unique<EngineContext>( in Compute()
209 const std::unique_ptr<EngineContext>& engine = pair.second; in Compute()
|
| D | trt_engine_resource_ops_test.cc | 160 absl::make_unique<EngineContext>(std::move(engine), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_lru_cache.h | 121 struct EngineContext { struct
122 EngineContext() {} // Creates an empty context. in EngineContext() argument
123 EngineContext(TrtUniquePtrType<nvinfer1::ICudaEngine>&& input_cuda_engine, in EngineContext() function
128 EngineContext(TrtUniquePtrType<nvinfer1::ICudaEngine>&& input_cuda_engine, in EngineContext() function
207 EngineContext* GetEngineContext(const std::vector<TensorShape>& input_shapes);
213 EngineContext* GetEngineContext(const int profile_id);
219 LRUCache<std::vector<TensorShape>, std::unique_ptr<EngineContext>,
|
| D | trt_lru_cache.cc | 99 EngineContext* TRTEngineCacheResource::GetEngineContext( in GetEngineContext()
101 EngineContext* engine_context = nullptr; in GetEngineContext()
122 EngineContext* TRTEngineCacheResource::GetEngineContext(const int profile_id) { in GetEngineContext()
|