Home
last modified time | relevance | path

Searched refs:allow_build_at_runtime_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dtrt_optimization_pass.h45 allow_build_at_runtime_(true) { in name_()
78 bool allow_build_at_runtime_; variable
Dtrt_optimization_pass.cc74 allow_build_at_runtime_ = params.at("allow_build_at_runtime").b(); in Init()
245 cp.allow_build_at_runtime = allow_build_at_runtime_; in Optimize()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc185 bool allow_build_at_runtime_; member in tensorflow::tensorrt::TRTEngineOp
355 context->GetAttr("_allow_build_at_runtime", &allow_build_at_runtime_); in TRTEngineOp()
360 allow_build_at_runtime_ = true; in TRTEngineOp()
902 if (!allow_build_at_runtime_) { in GetEngine()
978 if (!allow_build_at_runtime_) { in GetEngine()