Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h58 struct ConversionParams { struct
93 Status ConvertAfterShapes(const ConversionParams& params); argument
96 std::pair<int, Allocator*> GetDeviceAndAllocator(const ConversionParams& params,
Dconvert_graph_test.cc151 ConversionParams params; in TEST()
Dconvert_graph.cc400 Status CreateTRTNode(const ConversionParams& params, in CreateTRTNode()
730 std::pair<int, Allocator*> GetDeviceAndAllocator(const ConversionParams& params, in GetDeviceAndAllocator()
788 Status ConvertAfterShapes(const ConversionParams& params) { in ConvertAfterShapes()
Dtrt_optimization_pass.cc230 ConversionParams cp; in Optimize()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py59 ConversionParams = namedtuple("ConversionParams", [ variable
174 return ConversionParams(