Home
last modified time | relevance | path

Searched refs:MaybeUpdateBatchSize (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc535 Status MaybeUpdateBatchSize(int batch_size) { in MaybeUpdateBatchSize() function in tensorflow::tensorrt::convert::ConverterTest
536 return converter_->MaybeUpdateBatchSize(batch_size); in MaybeUpdateBatchSize()
765 TEST_F(ConverterTest, MaybeUpdateBatchSize) { in TEST_F() argument
768 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()
Dconvert_nodes.h520 Status MaybeUpdateBatchSize(int batch_size);
Dconvert_nodes.cc1056 Status status = MaybeUpdateBatchSize(batch_size); in AddInputTensor()
1118 Status Converter::MaybeUpdateBatchSize(int batch_size) { in MaybeUpdateBatchSize() function in tensorflow::tensorrt::convert::Converter