Searched refs:device_buffers_ (Results 1 – 3 of 3) sorted by relevance
167 device_buffers_(device_buffers) {} in ShardedDeviceArray()171 pybind11::list GetDeviceBuffers() const { return device_buffers_; } in GetDeviceBuffers()181 pybind11::list device_buffers_; variable
171 std::unordered_map<string, std::pair<void*, size_t>> device_buffers_; variable
1041 cres->device_buffers_.emplace( in AllocateCalibrationResources()1046 new TRTInt8Calibrator(cres->device_buffers_, batch_size, name())); in AllocateCalibrationResources()