Searched defs:use_implicit_batch (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_utils.cc | 76 int binding_index, bool use_implicit_batch, in GetTrtBindingShape()
137 std::vector<void*>& buffers, bool use_implicit_batch, in SetTrtEngineInputs()
220 bool use_implicit_batch, int batch_size, in SetTrtEngineOutputs()
281 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 | 48 bool use_implicit_batch = true; member
|
| D | convert_nodes.h | 381 const bool use_implicit_batch; member
500 bool use_implicit_batch() const { return use_implicit_batch_; } in use_implicit_batch() function
|
| D | convert_nodes.cc | 306 const bool use_implicit_batch, in ValidateTensorProperties()
363 const bool use_implicit_batch, in GetTrtBroadcastShape()
530 bool use_implicit_batch, int* trt_axis) { in ConvertAxis()
1105 bool use_implicit_batch) in OpConverterParams()
1139 bool use_implicit_batch) in TrtNodeValidator()
1244 nvinfer1::ILogger* trt_logger, const bool use_implicit_batch, in Create()
1255 const bool use_implicit_batch, in Converter()
6326 const bool use_implicit_batch, bool* convert_successfully, in ConvertGraphDefToEngine()
|
| D | convert_nodes_test.cc | 3493 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 | 237 use_implicit_batch=True): argument
|