Searched refs:cache_resource (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 162 TRTEngineCacheResource* cache_resource = nullptr; in TEST_F() local
164 device_->resource_manager()->Lookup("TF-TRT", "myop", &cache_resource)); in TEST_F()
165 core::ScopedUnref sc(cache_resource); in TEST_F()
168 auto cache = &cache_resource->cache_; in TEST_F()
215 TRTEngineCacheResource* cache_resource = nullptr; in TEST_F() local
217 device_->resource_manager()->Lookup("TF-TRT", "myop", &cache_resource)); in TEST_F()
218 core::ScopedUnref sc(cache_resource); in TEST_F()
222 auto cache = &cache_resource->cache_; in TEST_F()
243 TRTEngineCacheResource* cache_resource = nullptr; in TEST_F() local
245 device_->resource_manager()->Lookup("TF-TRT", "myop", &cache_resource)); in TEST_F()
[all …]
|
| D | trt_engine_op.cc | 139 OpKernelContext* ctx, TRTEngineCacheResource* cache_resource);
147 TRTEngineCacheResource* cache_resource);
830 TRTEngineCacheResource* cache_resource) { in BuildEngine() argument
845 conversion_input_shapes, &logger, cache_resource->allocator_.get(), in BuildEngine()
847 &cache_resource->profiles_, name()); in BuildEngine()
855 cache_resource->cache_.emplace(input_concrete_shapes, in BuildEngine()
|