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

  /external/tensorflow/tensorflow/contrib/tensorrt/python/
__init__.py 15 """Exposes the python wrapper for TensorRT graph transforms."""
22 from tensorflow.contrib.tensorrt.python.ops import trt_engine_op
23 from tensorflow.contrib.tensorrt.python.trt_convert import create_inference_graph
trt_convert.py 23 from tensorflow.contrib.tensorrt.wrap_conversion import trt_convert
  /external/tensorflow/tensorflow/contrib/tensorrt/log/
trt_logger.h 23 #include "tensorrt/include/NvInfer.h"
26 namespace tensorrt { namespace in namespace:tensorflow
36 } // namespace tensorrt
trt_logger.cc 16 #include "tensorflow/contrib/tensorrt/log/trt_logger.h"
23 namespace tensorrt { namespace in namespace:tensorflow
25 // Use TF logging for TensorRT informations
53 } // namespace tensorrt
  /external/tensorflow/tensorflow/contrib/tensorrt/
__init__.py 15 """Exposes the python wrapper for TensorRT graph transforms."""
22 from tensorflow.contrib.tensorrt.python import *
trt_conversion.i 61 #include "tensorflow/contrib/tensorrt/convert/convert_graph.h"
100 tensorflow::tensorrt::convert::ConvertGraphDefToTensorRT(
120 return std::pair<string, string>{"9;TensorRT is not enabled!", ""};
  /external/tensorflow/tensorflow/contrib/tensorrt/convert/
convert_graph.h 28 namespace tensorrt { namespace in namespace:tensorflow
41 } // namespace tensorrt
convert_nodes.h 32 namespace tensorrt { namespace in namespace:tensorflow
46 } // namespace tensorrt
convert_graph.cc 16 #include "tensorflow/contrib/tensorrt/convert/convert_graph.h"
24 #include "tensorflow/contrib/tensorrt/convert/convert_nodes.h"
25 #include "tensorflow/contrib/tensorrt/segment/segment.h"
44 #include "tensorrt/include/NvInfer.h"
47 namespace tensorrt { namespace in namespace:tensorflow
59 // LINT.ThenChange(//tensorflow/contrib/tensorrt/convert/convert_nodes.h)
150 // Build TensorRT node and add it to the graph
237 // Segment the graph into subgraphs that can be converted to TensorRT
238 tensorflow::tensorrt::segment::SegmentOptions segment_options;
247 tensorflow::tensorrt::segment::SegmentNodesVector segments
    [all...]
convert_nodes.cc 16 #include "tensorflow/contrib/tensorrt/convert/convert_nodes.h"
42 #include "tensorflow/contrib/tensorrt/log/trt_logger.h"
43 #include "tensorrt/include/NvInfer.h"
50 namespace tensorrt { namespace in namespace:tensorflow
    [all...]
  /external/tensorflow/tensorflow/contrib/tensorrt/segment/
segment.h 27 namespace tensorrt { namespace in namespace:tensorflow
38 // Get the subgraphs of a graph that can be handled by TensorRT.
42 // that node can be handled by TensorRT.
43 // @param segments Returns the TensorRT segments/subgraphs. Each entry
53 } // namespace tensorrt
union_find.h 20 namespace tensorrt { namespace in namespace:tensorflow
76 } // namespace tensorrt
segment.cc 16 #include "tensorflow/contrib/tensorrt/segment/segment.h"
22 #include "tensorflow/contrib/tensorrt/segment/union_find.h"
31 namespace tensorrt { namespace in namespace:tensorflow
252 } // namespace tensorrt
segment_test.cc 16 #include "tensorflow/contrib/tensorrt/segment/segment.h"
26 namespace tensorrt { namespace in namespace:tensorflow
366 } // namespace tensorrt
  /external/tensorflow/tensorflow/contrib/tensorrt/kernels/
trt_engine_op.h 28 #include "tensorrt/include/NvInfer.h"
31 namespace tensorrt { namespace in namespace:tensorflow
56 } // namespace tensorrt
trt_engine_op.cc 15 #include "tensorflow/contrib/tensorrt/kernels/trt_engine_op.h"
17 #include "tensorflow/contrib/tensorrt/log/trt_logger.h"
27 namespace tensorrt { namespace in namespace:tensorflow
28 static ::tensorflow::tensorrt::Logger logger;
136 } // namespace tensorrt
  /external/tensorflow/tensorflow/contrib/tensorrt/python/ops/
trt_engine_op.py 25 from tensorflow.contrib.tensorrt.ops.gen_trt_engine_op import *
  /external/tensorflow/tensorflow/contrib/tensorrt/shape_fn/
trt_shfn.cc 16 #include "tensorflow/contrib/tensorrt/shape_fn/trt_shfn.h"
23 #include "tensorflow/contrib/tensorrt/log/trt_logger.h"
25 #include "tensorrt/include/NvInfer.h"
31 tensorflow::tensorrt::Logger logger;
50 // TODO(jie): TensorRT engine requires consistent batch between inputs
52 LOG(FATAL) << "TensorRT engine requires consistent batch size";
76 LOG(FATAL) << "TensorRT engine cannot find binding: " << output_nodes[i];
  /external/tensorflow/tensorflow/contrib/tensorrt/test/
test_tftrt.py 15 """Script to test TF-TensorRT integration."""
27 from tensorflow.contrib import tensorrt as trt

Completed in 125 milliseconds