Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
H A Dtrt_shape_optimization_profiles.cc256 const std::vector<PartialTensorShape>& input_partial_shapes, in InitProfiles()
457 const std::vector<PartialTensorShape>& input_partial_shapes) { in SetShapeTensorMask()
H A Dtrt_shape_optimization_profiles_test.cc209 std::vector<PartialTensorShape> input_partial_shapes; in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
H A Dtrt_engine_resource_ops_test.cc200 std::vector<PartialTensorShape> input_partial_shapes; in CreateTRTEngine() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc1295 std::vector<PartialTensorShape> input_partial_shapes; in BuildAndRun() local