Home
last modified time | relevance | path

Searched defs:input_partial_shapes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles.cc101 const std::vector<PartialTensorShape>& input_partial_shapes) { in InitProfiles()
208 const std::vector<PartialTensorShape>& input_partial_shapes) { in SetShapeTensorMask()
Dtrt_shape_optimization_profiles_test.cc173 std::vector<PartialTensorShape> input_partial_shapes; in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1485 std::vector<PartialTensorShape> input_partial_shapes; in BuildAndRun() local