Home
last modified time | relevance | path

Searched refs:device_buffers_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_resources.h54 std::unordered_map<string, std::pair<void*, size_t>> device_buffers_; variable
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc678 cres->device_buffers_.emplace( in AllocateCalibrationResources()
683 new TRTInt8Calibrator(cres->device_buffers_, batch_size, name())); in AllocateCalibrationResources()