Searched refs:profile_generation_mode_ (Results 1 – 1 of 1) sorted by relevance
179 bool profile_generation_mode_; member in tensorflow::tensorrt::TRTEngineOp404 context->GetAttr("_profile_generation_mode", &profile_generation_mode_); in TRTEngineOp()409 profile_generation_mode_ = false; in TRTEngineOp()412 OP_REQUIRES(context, !profile_generation_mode_, in TRTEngineOp()675 if (profile_generation_mode_) { in ComputeAsync()