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

  /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/log/
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
trt_logger.h 23 #include "tensorrt/include/NvInfer.h"
26 namespace tensorrt { namespace in namespace:tensorflow
36 } // namespace tensorrt
  /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

Completed in 80 milliseconds