Searched refs:GetEngineCacheResource (Results 1 – 2 of 2) sorted by relevance
268 Status GetEngineCacheResource(const NodeDef& node, Session* session, in GetEngineCacheResource() function293 TF_RETURN_IF_ERROR(GetEngineCacheResource(node, session, &resource)); in ReadSerializedEngine()
178 Status GetEngineCacheResource(OpKernelContext* ctx,801 OP_REQUIRES_OK_ASYNC(ctx, GetEngineCacheResource(ctx, &cache_res), in ComputeAsync()1017 Status TRTEngineOp::GetEngineCacheResource(OpKernelContext* ctx, in GetEngineCacheResource() function in tensorflow::tensorrt::TRTEngineOp