Searched refs:trt_allocator (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | BUILD | 101 ":trt_allocator",
133 ":trt_allocator",
242 ":trt_allocator",
298 ":trt_allocator",
312 name = "trt_allocator",
313 srcs = ["utils/trt_allocator.cc"],
314 hdrs = ["utils/trt_allocator.h"],
331 ":trt_allocator",
398 ":trt_allocator",
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 437 std::unique_ptr<TRTBaseAllocator> trt_allocator; in CreateTRTNode() local
440 trt_allocator.reset(new TRTDeviceAllocator(device_allocator.second)); in CreateTRTNode()
456 trt_allocator.get(), /*calibrator=*/nullptr, &engine, in CreateTRTNode()
|