Searched refs:MaybeUpdateBatchSize (Results 1 – 3 of 3) sorted by relevance
535 Status MaybeUpdateBatchSize(int batch_size) { in MaybeUpdateBatchSize() function in tensorflow::tensorrt::convert::ConverterTest536 return converter_->MaybeUpdateBatchSize(batch_size); in MaybeUpdateBatchSize()765 TEST_F(ConverterTest, MaybeUpdateBatchSize) { in TEST_F() argument768 TF_EXPECT_OK(MaybeUpdateBatchSize(-1)); in TEST_F()771 TF_EXPECT_OK(MaybeUpdateBatchSize(123)); in TEST_F()774 TF_EXPECT_OK(MaybeUpdateBatchSize(123)); in TEST_F()777 TF_EXPECT_OK(MaybeUpdateBatchSize(-1)); in TEST_F()780 ExpectStatus(MaybeUpdateBatchSize(124), error::INVALID_ARGUMENT, in TEST_F()789 TF_EXPECT_OK(MaybeUpdateBatchSize(123)); in TEST_F()
520 Status MaybeUpdateBatchSize(int batch_size);
1056 Status status = MaybeUpdateBatchSize(batch_size); in AddInputTensor()1118 Status Converter::MaybeUpdateBatchSize(int batch_size) { in MaybeUpdateBatchSize() function in tensorflow::tensorrt::convert::Converter