Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc814 cudaError_t cuda_error = cudaGetDevice(&cuda_device_id); in ConvertAfterShapes() local
815 if (cuda_error != cudaSuccess) { in ConvertAfterShapes()
817 << cudaGetErrorString(cuda_error); in ConvertAfterShapes()