Home
last modified time | relevance | path

Searched defs:ConversionParams (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h58 struct ConversionParams { struct
59 const GraphDef* input_graph_def = nullptr;
60 const std::vector<string>* output_names = nullptr;
61 size_t max_batch_size = 1;
62 size_t max_workspace_size_bytes = 1 << 30;
63 GraphDef* output_graph_def = nullptr;
64 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32;
65 int minimum_segment_size = 3;
66 const grappler::GraphProperties* graph_properties = nullptr;
67 const grappler::Cluster* cluster = nullptr;
[all …]
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py59 ConversionParams = namedtuple("ConversionParams", [ variable