Searched defs:use_implicit_batch (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| H A D | trt_engine_utils.cc | 62 int binding_index, bool use_implicit_batch, in GetTrtBindingShape()
118 std::vector<void*>& buffers, bool use_implicit_batch, in SetTrtEngineInputs()
207 bool use_implicit_batch, int batch_size, in SetTrtEngineOutputs()
256 bool use_implicit_batch, int batch_size) { in TrtEnqueue()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 221 const bool use_implicit_batch, in ValidateTensorProperties()
279 const bool use_implicit_batch, in GetTrtBroadcastShape()
551 bool use_implicit_batch, int* trt_axis) { in ConvertAxis()
842 bool use_implicit_batch, bool use_explicit_precision) in OpConverterParams()
871 bool use_implicit_batch, bool use_explicit_precision) in TrtNodeValidator()
1021 nvinfer1::ILogger* trt_logger, const bool use_implicit_batch, in Create()
1033 const bool use_implicit_batch, in Converter()
5111 bool use_implicit_batch) { in ConvertDepthSpaceShuffle()
5141 bool use_implicit_batch) { in ConvertDepthSpaceShuffle()
5836 const bool use_implicit_batch, bool* convert_successfully, in ConvertGraphDefToEngine()
|
| H A D | trt_optimization_pass.h | 54 bool use_implicit_batch = true; member
|
| H A D | op_converter.h | 64 const bool use_implicit_batch; member
|
| H A D | convert_nodes.h | 317 bool use_implicit_batch() const { return use_implicit_batch_; } in use_implicit_batch() function
|
| H A D | convert_nodes_test.cc | 4965 const bool use_implicit_batch = (trt_mode_ == TrtTestMode::kImplicitBatch); in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| H A D | segment.h | 42 bool use_implicit_batch = true; member
|
| H A D | segment.cc | 641 bool use_implicit_batch) { in GetClusterBatchSizeForNode()
676 bool use_implicit_batch) { in AddSegmentForNode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_op_test.cc | 69 bool use_implicit_batch = true,
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| H A D | trt_convert.py | 282 use_implicit_batch=True, argument
|