Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py156 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
163 cls).__new__(cls, max_workspace_size_bytes, precision_mode,
291 conversion_params.max_workspace_size_bytes)
378 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
472 max_workspace_size_bytes=max_workspace_size_bytes,
1240 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
1309 max_workspace_size_bytes=max_workspace_size_bytes,
Dtrt_convert_windows.py83 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
Dtrt_convert_test.py205 max_workspace_size_bytes=TrtConvertTest._TRT_MAX_WORKSPACE_SIZE_BYTES,
381 max_workspace_size_bytes=10 << 20, # Use a smaller workspace. argument
388 max_workspace_size_bytes=max_workspace_size_bytes,
865 input_saved_model_dir, max_workspace_size_bytes=10 << 40)
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h94 max_workspace_size_bytes(0), in EngineInfo()
112 int64 max_workspace_size_bytes; member
149 size_t max_workspace_size_bytes,
480 int max_batch_size, size_t max_workspace_size_bytes,
Dconvert_graph.h38 size_t max_workspace_size_bytes = 1 << 30; member
Dconvert_graph.cc455 max_batch_size, info.max_workspace_size_bytes, input_shapes, trt_logger, in CreateTRTNode()
492 .Attr("workspace_size_bytes", info.max_workspace_size_bytes) in CreateTRTNode()
836 engine.max_workspace_size_bytes = params.max_workspace_size_bytes; in ConvertAfterShapes()
837 VLOG(1) << "Assigned " << engine.max_workspace_size_bytes << " bytes to " in ConvertAfterShapes()
Dconvert_graph_test.cc167 params.max_workspace_size_bytes = 8 << 20; in RunConvertAfterShape()
Dtrt_optimization_pass.cc236 cp.max_workspace_size_bytes = max_workspace_size_bytes_; in Optimize()
Dconvert_nodes.cc1451 size_t max_workspace_size_bytes, nvinfer1::IGpuAllocator* allocator, in BuildCudaEngine() argument
1467 builder_config->setMaxWorkspaceSize(max_workspace_size_bytes); in BuildCudaEngine()
1509 "Max-Workspace-Size:", max_workspace_size_bytes); in BuildCudaEngine()
1517 trt_builder_->setMaxWorkspaceSize(max_workspace_size_bytes); in BuildCudaEngine()
6321 size_t max_workspace_size_bytes, in ConvertGraphDefToEngine() argument
6447 engine, max_batch_size, max_workspace_size_bytes, allocator, calibrator, in ConvertGraphDefToEngine()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.experimental.tensorrt.-conversion-params.pbtxt11 name: "max_workspace_size_bytes"
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_cli.py782 max_workspace_size_bytes=args.max_workspace_size_bytes,
800 max_workspace_size_bytes=args.max_workspace_size_bytes,
/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
Dmodel_handler.py459 max_workspace_size_bytes=trt_convert_params.max_workspace_size_bytes,
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py280 max_workspace_size_bytes=1 << 25,
466 max_workspace_size_bytes=conversion_params.max_workspace_size_bytes,
Dquantization_mnist_test.py154 max_workspace_size_bytes=1 << 28,