Searched defs:use_implicit_batch (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_utils.cc | 47 int binding_index, bool use_implicit_batch, in GetTrtBindingShape()
99 std::vector<void*>& buffers, bool use_implicit_batch, in SetTrtEngineInputs()
182 bool use_implicit_batch, int batch_size, in SetTrtEngineOutputs()
245 bool use_implicit_batch, int batch_size) { in TrtEnqueue()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.h | 42 bool use_implicit_batch = true; member
|
| D | segment.cc | 636 bool use_implicit_batch) { in GetClusterBatchSizeForNode()
671 bool use_implicit_batch) { in AddSegmentForNode()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.h | 49 bool use_implicit_batch = true; member
|
| D | convert_nodes.h | 379 const bool use_implicit_batch; member
498 bool use_implicit_batch() const { return use_implicit_batch_; } in use_implicit_batch() function
|
| D | convert_nodes.cc | 300 const bool use_implicit_batch, in ValidateTensorProperties()
358 const bool use_implicit_batch, in GetTrtBroadcastShape()
611 bool use_implicit_batch, int* trt_axis) { in ConvertAxis()
1128 bool use_implicit_batch) in OpConverterParams()
1166 bool use_implicit_batch) in TrtNodeValidator()
1271 nvinfer1::ILogger* trt_logger, const bool use_implicit_batch, in Create()
1282 const bool use_implicit_batch, in Converter()
6249 bool use_implicit_batch) { in ConvertDepthSpaceShuffle()
6279 bool use_implicit_batch) { in ConvertDepthSpaceShuffle()
6803 const bool use_implicit_batch, bool* convert_successfully, in ConvertGraphDefToEngine()
|
| D | convert_nodes_test.cc | 3800 const bool use_implicit_batch = (trt_mode_ == TrtTestMode::kImplicitBatch); in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 64 bool use_implicit_batch = true,
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 275 use_implicit_batch=True, argument
|