Searched refs:CreateExecutionContexts (Results 1 – 5 of 5) sorted by relevance
145 profile.CreateExecutionContexts(engine.get(), exec_context_, nullptr)); in TEST_F()185 profile.CreateExecutionContexts(engine.get(), exec_context_, nullptr)); in TEST_F()
161 Status CreateExecutionContexts(nvinfer1::ICudaEngine* engine,
233 Status TrtShapeOptimizationProfile::CreateExecutionContexts( in CreateExecutionContexts() function in tensorflow::tensorrt::TrtShapeOptimizationProfile
148 OP_REQUIRES_OK(ctx, resource->profiles_.CreateExecutionContexts( in Compute()
1000 TF_RETURN_IF_ERROR(cache_res->profiles_.CreateExecutionContexts( in GetEngine()