Searched refs:input_partial_shapes (Results 1 – 5 of 5) sorted by relevance
256 const std::vector<PartialTensorShape>& input_partial_shapes, in InitProfiles() argument310 SetShapeTensorMask(input_partial_shapes); in InitProfiles()311 if (input_partial_shapes.size() > 0) { in InitProfiles()315 for (int i = 0; i < input_partial_shapes.size(); i++) { in InitProfiles()316 auto network_input = input_partial_shapes[i]; in InitProfiles()457 const std::vector<PartialTensorShape>& input_partial_shapes) { in SetShapeTensorMask() argument458 if (is_shape_tensor_.size() == input_partial_shapes.size()) { in SetShapeTensorMask()462 is_shape_tensor_.resize(input_partial_shapes.size(), false); in SetShapeTensorMask()463 for (int i = 0; i < input_partial_shapes.size(); i++) { in SetShapeTensorMask()464 is_shape_tensor_[i] = IsTrtShapeTensorCompatible(input_partial_shapes[i]); in SetShapeTensorMask()
209 std::vector<PartialTensorShape> input_partial_shapes; in TEST_P() local210 TF_CHECK_OK(GetNetworkInputShapes(network_.get(), &input_partial_shapes)); in TEST_P()211 profile.InitProfiles(input_partial_shapes, strategy_); in TEST_P()
251 void InitProfiles(const std::vector<PartialTensorShape>& input_partial_shapes,330 const std::vector<PartialTensorShape>& input_partial_shapes);
200 std::vector<PartialTensorShape> input_partial_shapes; in CreateTRTEngine() local201 TF_CHECK_OK(GetNetworkInputShapes(network.get(), &input_partial_shapes)); in CreateTRTEngine()202 profile.InitProfiles(input_partial_shapes, in CreateTRTEngine()
1295 std::vector<PartialTensorShape> input_partial_shapes; in BuildAndRun() local1297 GetNetworkInputShapes(converter_->network(), &input_partial_shapes)); in BuildAndRun()1298 profiles.InitProfiles(input_partial_shapes, in BuildAndRun()