Home
last modified time | relevance | path

Searched refs:max_workspace_size_bytes (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensorrt/python/
Dtrt_convert.py28 max_workspace_size_bytes=trt_convert.DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
42 max_workspace_size_bytes=max_workspace_size_bytes,
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py465 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
547 "max_workspace_size_bytes"].i = max_workspace_size_bytes
565 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
702 self._max_workspace_size_bytes = max_workspace_size_bytes
713 max_workspace_size_bytes=self._max_workspace_size_bytes,
785 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
866 max_workspace_size_bytes=max_workspace_size_bytes,
Dtrt_convert_test.py67 max_workspace_size_bytes=1234,
186 max_workspace_size_bytes=TrtConvertTest._TRT_MAX_WORKSPACE_SIZE_BYTES,
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h62 size_t max_workspace_size_bytes = 1 << 30; member
85 size_t max_batch_size, size_t max_workspace_size_bytes,
Dconvert_nodes.h100 max_workspace_size_bytes(0), in EngineInfo()
115 int64 max_workspace_size_bytes; member
151 size_t max_workspace_size_bytes,
Dconvert_graph.cc103 size_t max_batch_size, size_t max_workspace_size_bytes, in ConvertGraphDefToTensorRT() argument
162 parameters["max_workspace_size_bytes"].set_i(max_workspace_size_bytes); in ConvertGraphDefToTensorRT()
518 max_batch_size, info.max_workspace_size_bytes, input_shapes, in CreateTRTNode()
566 .Attr("workspace_size_bytes", info.max_workspace_size_bytes) in CreateTRTNode()
900 engine.max_workspace_size_bytes = in ConvertAfterShapes()
901 params.max_workspace_size_bytes * in ConvertAfterShapes()
Dtrt_optimization_pass.cc258 cp.max_workspace_size_bytes = max_workspace_size_bytes_; in Optimize()
Dconvert_nodes.cc4057 size_t max_workspace_size_bytes, in ConvertGraphDefToEngine() argument
4069 builder->setMaxWorkspaceSize(max_workspace_size_bytes); in ConvertGraphDefToEngine()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py179 max_workspace_size_bytes=1 << 25,
224 conversion_params.max_workspace_size_bytes,
306 max_workspace_size_bytes=conversion_params.max_workspace_size_bytes,
Dquantization_mnist_test.py149 max_workspace_size_bytes=1 << 28,
/external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/
Dplugin_test.py87 max_workspace_size_bytes=1 << 25,
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_cli.py674 max_workspace_size_bytes=args.max_workspace_size_bytes,