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.h618 Status MaybeUpdateBatchSize(int batch_size);
Dconvert_nodes_test.cc753 Status MaybeUpdateBatchSize(int batch_size) { in MaybeUpdateBatchSize() function in tensorflow::tensorrt::convert::ConverterTest
754 return converter_->MaybeUpdateBatchSize(batch_size); in MaybeUpdateBatchSize()
1014 TEST_F(ConverterTest, MaybeUpdateBatchSize) { in TEST_F() argument
1017 TF_EXPECT_OK(MaybeUpdateBatchSize(-1)); in TEST_F()
1020 TF_EXPECT_OK(MaybeUpdateBatchSize(123)); in TEST_F()
1023 TF_EXPECT_OK(MaybeUpdateBatchSize(123)); in TEST_F()
1026 TF_EXPECT_OK(MaybeUpdateBatchSize(-1)); in TEST_F()
1029 ExpectStatus(MaybeUpdateBatchSize(124), error::INVALID_ARGUMENT, in TEST_F()
1038 TF_EXPECT_OK(MaybeUpdateBatchSize(123)); in TEST_F()
Dconvert_nodes.cc1339 status = MaybeUpdateBatchSize(batch_size); in AddInputTensor()
1542 Status Converter::MaybeUpdateBatchSize(int batch_size) { in MaybeUpdateBatchSize() function in tensorflow::tensorrt::convert::Converter