Home
last modified time | relevance | path

Searched refs:allow_dynamic_non_batch_dim (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.h48 bool allow_dynamic_non_batch_dim = false; member
Dsegment.cc688 if (!options.use_implicit_batch && !options.allow_dynamic_non_batch_dim) { in SegmentGraph()
697 if (!options.allow_dynamic_non_batch_dim && !graph_properties) { in SegmentGraph()
760 } else if (!options.allow_dynamic_non_batch_dim && in SegmentGraph()
Dsegment_test.cc108 segment_options_.allow_dynamic_non_batch_dim = true; in DisableImplicitBatchMode()
114 segment_options_.allow_dynamic_non_batch_dim = false; in EnableImplicitBatchModeForStaticEngine()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc738 segment_options.allow_dynamic_non_batch_dim = in ConvertAfterShapes()