OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:maximum_cached_engines
(Results
1 - 9
of
9
) sorted by null
/external/tensorflow/tensorflow/contrib/tensorrt/python/
trt_convert.py
32
maximum_cached_engines
=1,
46
maximum_cached_engines
=
maximum_cached_engines
,
/external/tensorflow/tensorflow/python/compiler/tensorrt/
trt_convert.py
469
maximum_cached_engines
=1,
487
maximum_cached_engines
: max number of cached TRT engines in dynamic TRT
493
should be <=
maximum_cached_engines
, and the dynamic TRT op will use
549
optimizer.parameter_map["
maximum_cached_engines
"].i =
maximum_cached_engines
569
maximum_cached_engines
=1,
598
maximum_cached_engines
: max number of cached TRT engines in dynamic TRT
604
should be <=
maximum_cached_engines
, and the dynamic TRT op will use
681
if len(cached_engine_batches) >
maximum_cached_engines
:
683
"
maximum_cached_engines
items."
[
all
...]
trt_convert_test.py
71
maximum_cached_engines
=2,
85
"max_workspace_size_bytes", "precision_mode", "
maximum_cached_engines
",
97
self.assertEqual(2, trt_optimizer.parameter_map["
maximum_cached_engines
"].i)
176
maximum_cached_engines
=1,
191
maximum_cached_engines
=
maximum_cached_engines
,
383
maximum_cached_engines
=2,
421
maximum_cached_engines
=2, # This is noop, added just for testing.
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
dynamic_input_shapes_test.py
85
maximum_cached_engines
=10,
int32_test.py
65
maximum_cached_engines
=1,
tf_trt_integration_test_base.py
61
"minimum_segment_size", "is_dynamic_op", "
maximum_cached_engines
",
183
maximum_cached_engines
=1,
228
conversion_params.
maximum_cached_engines
,
310
maximum_cached_engines
=conversion_params.
maximum_cached_engines
,
322
assert conversion_params.
maximum_cached_engines
== 1
biasadd_matmul_test.py
124
maximum_cached_engines
=1,
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
convert_nodes.h
116
int
maximum_cached_engines
;
member in class:tensorflow::tensorrt::convert::EngineInfo::EngineType
convert_graph.cc
165
parameters["
maximum_cached_engines
"].set_i(max_cached_engines);
565
.Attr("max_cached_engines_count", info.
maximum_cached_engines
)
[
all
...]
Completed in 805 milliseconds