Home
last modified time | relevance | path

Searched defs:cuda_device_id (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_graph.cc644 int cuda_device_id = -1; in GetDeviceAndAllocator() local
699 int cuda_device_id = 0; in CreateStaticEngine() local
875 int cuda_device_id; in ConvertGraph() local