HomeSort by relevance Sort by last modified time
    Searched refs:trt_engine (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/tensorrt/shape_fn/
trt_shfn.cc 35 nvinfer1::ICudaEngine* trt_engine = infer->deserializeCudaEngine( local
66 int binding_index = trt_engine->getBindingIndex(output_nodes[i].c_str());
71 auto dims = trt_engine->getBindingDimensions(binding_index);
  /external/tensorflow/tensorflow/contrib/tensorrt/convert/
convert_nodes.cc     [all...]

Completed in 63 milliseconds