Searched refs:input_partial_shapes (Results 1 – 4 of 4) sorted by relevance
101 const std::vector<PartialTensorShape>& input_partial_shapes) { in InitProfiles() argument120 SetShapeTensorMask(input_partial_shapes); in InitProfiles()121 if (input_partial_shapes.size() > 0) { in InitProfiles()125 for (int i = 0; i < input_partial_shapes.size(); i++) { in InitProfiles()126 auto network_input = input_partial_shapes[i]; in InitProfiles()208 const std::vector<PartialTensorShape>& input_partial_shapes) { in SetShapeTensorMask() argument209 is_shape_tensor_.resize(input_partial_shapes.size(), false); in SetShapeTensorMask()210 for (int i = 0; i < input_partial_shapes.size(); i++) { in SetShapeTensorMask()211 is_shape_tensor_[i] = IsTrtShapeTensorCompatible(input_partial_shapes[i]); in SetShapeTensorMask()
173 std::vector<PartialTensorShape> input_partial_shapes; in TEST_F() local174 TF_CHECK_OK(GetNetworkInputShapes(network_.get(), &input_partial_shapes)); in TEST_F()175 profile.InitProfiles(input_partial_shapes); in TEST_F()
170 const std::vector<PartialTensorShape>& input_partial_shapes);210 const std::vector<PartialTensorShape>& input_partial_shapes);
1485 std::vector<PartialTensorShape> input_partial_shapes; in BuildAndRun() local1487 GetNetworkInputShapes(converter_->network(), &input_partial_shapes)); in BuildAndRun()1488 profiles.InitProfiles(input_partial_shapes); in BuildAndRun()