Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h33 struct ConversionParams { struct
53 Status ConvertAfterShapes(const ConversionParams& params); argument
56 std::pair<int, Allocator*> GetDeviceAndAllocator(const ConversionParams& params,
Dconvert_graph_test.cc78 ConversionParams params; in TEST()
164 ConversionParams params; in RunConvertAfterShape()
Dconvert_graph.cc83 EngineInfo::EngineType GetEngineType(const ConversionParams& params) { in GetEngineType()
91 bool AllowDynamicNonBatchDimension(const ConversionParams& params) { in AllowDynamicNonBatchDimension()
331 Status CreateTRTNode(const ConversionParams& params, in CreateTRTNode()
631 std::pair<int, Allocator*> GetDeviceAndAllocator(const ConversionParams& params, in GetDeviceAndAllocator()
682 Status ConvertAfterShapes(const ConversionParams& params) { in ConvertAfterShapes()
Dtrt_optimization_pass.cc231 ConversionParams cp; in Optimize()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.experimental.tensorrt.pbtxt4 name: "ConversionParams"
Dtensorflow.experimental.tensorrt.-conversion-params.pbtxt1 path: "tensorflow.experimental.tensorrt.ConversionParams"