Home
last modified time | relevance | path

Searched refs:allow_build_at_runtime (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h49 bool allow_build_at_runtime = true; member
Dconvert_nodes.h99 allow_build_at_runtime(true) {} in EngineInfo()
117 bool allow_build_at_runtime; member
Dconvert_graph.cc497 .Attr("_allow_build_at_runtime", info.allow_build_at_runtime) in CreateTRTNode()
781 curr_engine.allow_build_at_runtime = params.allow_build_at_runtime; in ConvertAfterShapes()
Dtrt_optimization_pass.cc245 cp.allow_build_at_runtime = allow_build_at_runtime_; in Optimize()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.experimental.tensorrt.-conversion-params.pbtxt7 name: "allow_build_at_runtime"
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op_test.cc65 bool allow_build_at_runtime = true) { argument
114 .Attr("_allow_build_at_runtime", allow_build_at_runtime)
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py161 allow_build_at_runtime=True): argument
165 use_calibration, allow_build_at_runtime)
300 "allow_build_at_runtime"].b = conversion_params.allow_build_at_runtime
477 allow_build_at_runtime=True)
1212 if not self._conversion_params.allow_build_at_runtime: