Home
last modified time | relevance | path

Searched defs:use_implicit_batch (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h51 bool use_implicit_batch = true; member
Dconvert_nodes.h368 const bool use_implicit_batch; member
485 bool use_implicit_batch() const { return use_implicit_batch_; } in use_implicit_batch() function
Dconvert_nodes.cc254 const bool use_implicit_batch, in ValidateTensorProperties()
309 const bool use_implicit_batch, in GetTrtBroadcastShape()
461 bool use_implicit_batch, int* trt_axis) { in ConvertAxis()
993 bool use_implicit_batch) in OpConverterParams()
1027 bool use_implicit_batch) in TrtNodeValidator()
1167 nvinfer1::ILogger* trt_logger, const bool use_implicit_batch) { in Create()
1176 const bool use_implicit_batch) in Converter()
5687 const bool use_implicit_batch, bool* convert_successfully) { in ConvertGraphDefToEngine()