tensor2tensor/problems_test.py::ProblemsTest::testImport
tensor2tensor/problems_test.py::ProblemsTest::test_session
tensor2tensor/bin/t2t_trainer_test.py::TrainerTest::testTrain
tensor2tensor/bin/t2t_trainer_test.py::TrainerTest::test_session
tensor2tensor/data_generators/algorithmic_math_test.py::AlgorithmicMathTest::testAlgebraInverse
tensor2tensor/data_generators/algorithmic_math_test.py::AlgorithmicMathTest::testAlgebraSimplify
tensor2tensor/data_generators/algorithmic_math_test.py::AlgorithmicMathTest::testCalculusIntegrate
tensor2tensor/data_generators/algorithmic_math_test.py::AlgorithmicMathTest::test_session
tensor2tensor/data_generators/algorithmic_test.py::AlgorithmicTest::testAdditionGenerator
tensor2tensor/data_generators/algorithmic_test.py::AlgorithmicTest::testIdentityGenerator
tensor2tensor/data_generators/algorithmic_test.py::AlgorithmicTest::testLowerEndianToNumber
tensor2tensor/data_generators/algorithmic_test.py::AlgorithmicTest::testMultiplicationGenerator
tensor2tensor/data_generators/algorithmic_test.py::AlgorithmicTest::testNumberToLowerEndian
tensor2tensor/data_generators/algorithmic_test.py::AlgorithmicTest::testReverseGenerator
tensor2tensor/data_generators/algorithmic_test.py::AlgorithmicTest::testReverseGeneratorNlpLike
tensor2tensor/data_generators/algorithmic_test.py::AlgorithmicTest::testSortGenerator
tensor2tensor/data_generators/algorithmic_test.py::AlgorithmicTest::testZipfDistribution
tensor2tensor/data_generators/algorithmic_test.py::AlgorithmicTest::test_session
tensor2tensor/data_generators/allen_brain_test.py::TestAllenBrain::test_generate_data_produces_examples_of_correct_shape
tensor2tensor/data_generators/allen_brain_test.py::TestAllenBrain::test_generator_produces_examples
tensor2tensor/data_generators/allen_brain_test.py::TestAllenBrain::test_session
tensor2tensor/data_generators/allen_brain_test.py::TestAllenBrain::test_transformer2d_single_step_e2e
tensor2tensor/data_generators/allen_brain_test.py::TestImageMock::test_image_mock_produces_expected_shape
tensor2tensor/data_generators/allen_brain_test.py::TestImageMock::test_session
tensor2tensor/data_generators/allen_brain_test.py::TestMockRawData::test_runs
tensor2tensor/data_generators/allen_brain_test.py::TestMockRawData::test_session
tensor2tensor/data_generators/audio_test.py::AudioTest::testDataCollection
tensor2tensor/data_generators/audio_test.py::AudioTest::test_session
tensor2tensor/data_generators/celeba_test.py::CelebaTest::testCelebaMultiResolutionPreprocessExampleArea
tensor2tensor/data_generators/celeba_test.py::CelebaTest::testCelebaMultiResolutionPreprocessExampleDefault
tensor2tensor/data_generators/celeba_test.py::CelebaTest::testCelebaMultiResolutionPreprocessExampleDilated
tensor2tensor/data_generators/celeba_test.py::CelebaTest::test_session
tensor2tensor/data_generators/common_voice_test.py::CommonVoiceTest::testCollectData
tensor2tensor/data_generators/common_voice_test.py::CommonVoiceTest::test_session
tensor2tensor/data_generators/desc2code_test.py::Desc2codeTest::testCppPreprocess
tensor2tensor/data_generators/desc2code_test.py::Desc2codeTest::test_session
tensor2tensor/data_generators/dna_encoder_test.py::DnaEncoderTest::test_delimited_dna_encoder
tensor2tensor/data_generators/dna_encoder_test.py::DnaEncoderTest::test_encode_decode
tensor2tensor/data_generators/dna_encoder_test.py::DnaEncoderTest::test_session
tensor2tensor/data_generators/gene_expression_test.py::GeneticsTest::testGenerateShardArgs
tensor2tensor/data_generators/gene_expression_test.py::GeneticsTest::testRecordToExample
tensor2tensor/data_generators/gene_expression_test.py::GeneticsTest::test_session
tensor2tensor/data_generators/generator_utils_test.py::GeneratorUtilsTest::testDatasetPacking
tensor2tensor/data_generators/generator_utils_test.py::GeneratorUtilsTest::testGenerateFiles
tensor2tensor/data_generators/generator_utils_test.py::GeneratorUtilsTest::testGetOrGenerateTxtVocab
tensor2tensor/data_generators/generator_utils_test.py::GeneratorUtilsTest::testGunzipFile
tensor2tensor/data_generators/generator_utils_test.py::GeneratorUtilsTest::testMaybeDownload
tensor2tensor/data_generators/generator_utils_test.py::GeneratorUtilsTest::testMaybeDownloadFromDrive
tensor2tensor/data_generators/generator_utils_test.py::GeneratorUtilsTest::testPacking
tensor2tensor/data_generators/generator_utils_test.py::GeneratorUtilsTest::test_session
tensor2tensor/data_generators/gym_env_test.py::GymEnvTest::test_channels
tensor2tensor/data_generators/gym_env_test.py::GymEnvTest::test_clipping
tensor2tensor/data_generators/gym_env_test.py::GymEnvTest::test_frame_numbers_are_continuous
tensor2tensor/data_generators/gym_env_test.py::GymEnvTest::test_generates_data
tensor2tensor/data_generators/gym_env_test.py::GymEnvTest::test_generating_and_loading_preserves_rollouts
tensor2tensor/data_generators/gym_env_test.py::GymEnvTest::test_no_resize_option
tensor2tensor/data_generators/gym_env_test.py::GymEnvTest::test_resize
tensor2tensor/data_generators/gym_env_test.py::GymEnvTest::test_session
tensor2tensor/data_generators/gym_env_test.py::GymEnvTest::test_shards_per_epoch
tensor2tensor/data_generators/gym_env_test.py::GymEnvTest::test_split_preserves_number_of_frames
tensor2tensor/data_generators/gym_env_test.py::GymEnvTest::test_split_preserves_number_of_rollouts
tensor2tensor/data_generators/gym_env_test.py::GymEnvTest::test_splits_dataset
tensor2tensor/data_generators/image_utils_test.py::ImageTest::testConvertPredictionsToImageSummaries
tensor2tensor/data_generators/image_utils_test.py::ImageTest::testImageAugmentation
tensor2tensor/data_generators/image_utils_test.py::ImageTest::testImageGenerator
tensor2tensor/data_generators/image_utils_test.py::ImageTest::testImageToSummaryValue
tensor2tensor/data_generators/image_utils_test.py::ImageTest::testMakeMultiscaleDilatedDivisible
tensor2tensor/data_generators/image_utils_test.py::ImageTest::testMakeMultiscaleDilatedIndivisible
tensor2tensor/data_generators/image_utils_test.py::ImageTest::testMakeMultiscaleDilatedLarger
tensor2tensor/data_generators/image_utils_test.py::ImageTest::testMakeMultiscaleDivisible
tensor2tensor/data_generators/image_utils_test.py::ImageTest::testMakeMultiscaleIndivisible
tensor2tensor/data_generators/image_utils_test.py::ImageTest::testMakeMultiscaleLarger
tensor2tensor/data_generators/image_utils_test.py::ImageTest::testRandomShift
tensor2tensor/data_generators/image_utils_test.py::ImageTest::test_session
tensor2tensor/data_generators/imagenet_test.py::ImagenetTest::testImagenetIsNormalized
tensor2tensor/data_generators/imagenet_test.py::ImagenetTest::testImagenetMultiResolutionPreprocessExampleArea
tensor2tensor/data_generators/imagenet_test.py::ImagenetTest::testImagenetMultiResolutionPreprocessExampleDefault
tensor2tensor/data_generators/imagenet_test.py::ImagenetTest::testImagenetMultiResolutionPreprocessExampleDilated
tensor2tensor/data_generators/imagenet_test.py::ImagenetTest::test_session
tensor2tensor/data_generators/mscoco_test.py::MscocoTest::testMsCocoMultiResolutionPreprocessExampleArea
tensor2tensor/data_generators/mscoco_test.py::MscocoTest::testMsCocoMultiResolutionPreprocessExampleDefault
tensor2tensor/data_generators/mscoco_test.py::MscocoTest::testMsCocoMultiResolutionPreprocessExampleDilated
tensor2tensor/data_generators/mscoco_test.py::MscocoTest::test_session
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_categorical_case0
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_categorical_case1
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_encode_decode_schedule0
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_encode_decode_schedule1
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_encode_decode_schedule2
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_get_multi_dataset0
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_get_multi_dataset1
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_get_multi_dataset2
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_get_schedule_distribution0
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_get_schedule_distribution1
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_linear_interpolation0
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_linear_interpolation1
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_multi_problem_v20
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_multi_problem_v21
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_session
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_step_interpolation0
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_step_interpolation1
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_tuplize0
tensor2tensor/data_generators/multi_problem_v2_test.py::MultiProblemV2Test::test_tuplize1
tensor2tensor/data_generators/paraphrase_ms_coco_test.py::ParaphraseGenerationProblemTest::testBidirectionalFalse
tensor2tensor/data_generators/paraphrase_ms_coco_test.py::ParaphraseGenerationProblemTest::testBidirectionalTrue
tensor2tensor/data_generators/paraphrase_ms_coco_test.py::ParaphraseGenerationProblemTest::testCombinationPairs
tensor2tensor/data_generators/paraphrase_ms_coco_test.py::ParaphraseGenerationProblemTest::test_session
tensor2tensor/data_generators/problem_test.py::ProblemTest::testDataFilenames
tensor2tensor/data_generators/problem_test.py::ProblemTest::testInputAndTargetModalitiesAreReversed
tensor2tensor/data_generators/problem_test.py::ProblemTest::testInputAndTargetVocabSizesAreReversed
tensor2tensor/data_generators/problem_test.py::ProblemTest::testNoShuffleDeterministic
tensor2tensor/data_generators/problem_test.py::ProblemTest::testNoShufflePreprocess
tensor2tensor/data_generators/problem_test.py::ProblemTest::testProblemHparamsInputOnlyModality
tensor2tensor/data_generators/problem_test.py::ProblemTest::testProblemHparamsModality
tensor2tensor/data_generators/problem_test.py::ProblemTest::testProblemHparamsTargetOnlyModality
tensor2tensor/data_generators/problem_test.py::ProblemTest::testServingInputFnUseTpu
tensor2tensor/data_generators/problem_test.py::ProblemTest::test_session
tensor2tensor/data_generators/program_search_test.py::ProgramSearchAlgolispTest::testEndToEnd
tensor2tensor/data_generators/program_search_test.py::ProgramSearchAlgolispTest::test_session
tensor2tensor/data_generators/style_transfer_test.py::StyleTransferProblemShakespeareTest::testSourceAndTargetPathsDevModern2Shakespeare
tensor2tensor/data_generators/style_transfer_test.py::StyleTransferProblemShakespeareTest::testSourceAndTargetPathsDevShakespeare2Modern
tensor2tensor/data_generators/style_transfer_test.py::StyleTransferProblemShakespeareTest::testSourceAndTargetPathsTrainModern2Shakespeare
tensor2tensor/data_generators/style_transfer_test.py::StyleTransferProblemShakespeareTest::testSourceAndTargetPathsTrainShakespeare2Modern
tensor2tensor/data_generators/style_transfer_test.py::StyleTransferProblemShakespeareTest::test_session
tensor2tensor/data_generators/text_encoder_test.py::NativeToUnicodeTest::test_native_to_unicode
tensor2tensor/data_generators/text_encoder_test.py::NativeToUnicodeTest::test_session
tensor2tensor/data_generators/text_encoder_test.py::EscapeUnescapeTokenTest::test_escape_token
tensor2tensor/data_generators/text_encoder_test.py::EscapeUnescapeTokenTest::test_session
tensor2tensor/data_generators/text_encoder_test.py::EscapeUnescapeTokenTest::test_unescape_token
tensor2tensor/data_generators/text_encoder_test.py::TokenTextEncoderTest::test_reserved_tokens_in_corpus
tensor2tensor/data_generators/text_encoder_test.py::TokenTextEncoderTest::test_save_and_reload
tensor2tensor/data_generators/text_encoder_test.py::TokenTextEncoderTest::test_session
tensor2tensor/data_generators/text_encoder_test.py::SubwordTextEncoderTest::test_build_from_generator
tensor2tensor/data_generators/text_encoder_test.py::SubwordTextEncoderTest::test_custom_reserved_tokens
tensor2tensor/data_generators/text_encoder_test.py::SubwordTextEncoderTest::test_encodable_when_not_in_alphabet
tensor2tensor/data_generators/text_encoder_test.py::SubwordTextEncoderTest::test_encode_decode
tensor2tensor/data_generators/text_encoder_test.py::SubwordTextEncoderTest::test_load_from_file
tensor2tensor/data_generators/text_encoder_test.py::SubwordTextEncoderTest::test_long_tokens
tensor2tensor/data_generators/text_encoder_test.py::SubwordTextEncoderTest::test_raises_exception_when_not_encodable
tensor2tensor/data_generators/text_encoder_test.py::SubwordTextEncoderTest::test_reserved_token_chars_not_in_alphabet
tensor2tensor/data_generators/text_encoder_test.py::SubwordTextEncoderTest::test_save_and_reload
tensor2tensor/data_generators/text_encoder_test.py::SubwordTextEncoderTest::test_save_and_reload_no_single_quotes
tensor2tensor/data_generators/text_encoder_test.py::SubwordTextEncoderTest::test_session
tensor2tensor/data_generators/text_encoder_test.py::SubwordTextEncoderTest::test_small_vocab
tensor2tensor/data_generators/text_encoder_test.py::SubwordTextEncoderTest::test_unicode
tensor2tensor/data_generators/text_encoder_test.py::OneHotClassLabelEncoderTest::test_one_hot_decode
tensor2tensor/data_generators/text_encoder_test.py::OneHotClassLabelEncoderTest::test_one_hot_encode
tensor2tensor/data_generators/text_encoder_test.py::OneHotClassLabelEncoderTest::test_session
tensor2tensor/data_generators/text_problems_test.py::TextProblems::testText2ClassTxtIterator
tensor2tensor/data_generators/text_problems_test.py::TextProblems::testText2ClassTxtIteratorWithStrs
tensor2tensor/data_generators/text_problems_test.py::TextProblems::testText2RealTxtIterator
tensor2tensor/data_generators/text_problems_test.py::TextProblems::testText2SelfTxtIterator
tensor2tensor/data_generators/text_problems_test.py::TextProblems::testText2TextTmpDir
tensor2tensor/data_generators/text_problems_test.py::TextProblems::testText2TextTxtIterator
tensor2tensor/data_generators/text_problems_test.py::TextProblems::testText2TextTxtTabIterator
tensor2tensor/data_generators/text_problems_test.py::TextProblems::testTxtLineIterator
tensor2tensor/data_generators/text_problems_test.py::TextProblems::test_session
tensor2tensor/data_generators/text_problems_test.py::DistributedText2TextProblemsTest::testInputShardingNoGeneratePerSplit
tensor2tensor/data_generators/text_problems_test.py::DistributedText2TextProblemsTest::testInputShardingWithGeneratePerSplit
tensor2tensor/data_generators/text_problems_test.py::DistributedText2TextProblemsTest::testOutputSharding
tensor2tensor/data_generators/text_problems_test.py::DistributedText2TextProblemsTest::testVocabularyIsAllTrain
tensor2tensor/data_generators/text_problems_test.py::DistributedText2TextProblemsTest::test_session
tensor2tensor/data_generators/timeseries_data_generator_test.py::TimeseriesDataGeneratorTest::testGenerateData
tensor2tensor/data_generators/timeseries_data_generator_test.py::TimeseriesDataGeneratorTest::test_session
tensor2tensor/data_generators/timeseries_test.py::TimeseriesTest::testTimeseriesSyntheticData10Series100kSamples
tensor2tensor/data_generators/timeseries_test.py::TimeseriesTest::testTimeseriesToyProblem
tensor2tensor/data_generators/timeseries_test.py::TimeseriesTest::testTimeseriesToyProblemNoInputs
tensor2tensor/data_generators/timeseries_test.py::TimeseriesTest::test_session
tensor2tensor/data_generators/tokenizer_test.py::TokenizerTest::test_decode
tensor2tensor/data_generators/tokenizer_test.py::TokenizerTest::test_encode
tensor2tensor/data_generators/tokenizer_test.py::TokenizerTest::test_invertibility_on_random_strings
tensor2tensor/data_generators/tokenizer_test.py::TokenizerTest::test_session
tensor2tensor/data_generators/tokenizer_test.py::TestTokenCounts::test_corpus_token_counts_no_split_on_newlines
tensor2tensor/data_generators/tokenizer_test.py::TestTokenCounts::test_corpus_token_counts_no_split_with_max_lines
tensor2tensor/data_generators/tokenizer_test.py::TestTokenCounts::test_corpus_token_counts_split_on_newlines
tensor2tensor/data_generators/tokenizer_test.py::TestTokenCounts::test_corpus_token_counts_split_with_max_lines
tensor2tensor/data_generators/tokenizer_test.py::TestTokenCounts::test_session
tensor2tensor/data_generators/tokenizer_test.py::TestTokenCounts::test_vocab_token_counts
tensor2tensor/data_generators/tokenizer_test.py::TestTokenCounts::test_vocab_token_counts_with_max_lines
tensor2tensor/data_generators/transduction_problems_test.py::TransductionProblem::testTransductionCopySequence
tensor2tensor/data_generators/transduction_problems_test.py::TransductionProblem::testTransductionCopySequenceSmall
tensor2tensor/data_generators/transduction_problems_test.py::TransductionProblem::testTransductionFlipBiGramSequence
tensor2tensor/data_generators/transduction_problems_test.py::TransductionProblem::testTransductionReverseSequence
tensor2tensor/data_generators/transduction_problems_test.py::TransductionProblem::testTransductionReverseSequenceSmall
tensor2tensor/data_generators/translate_ende_test.py::TranslateEndeTest::test_additional_datasets
tensor2tensor/data_generators/translate_ende_test.py::TranslateEndeTest::test_session
tensor2tensor/data_generators/translate_ende_test.py::TranslateEndeTest::test_source_data_files
tensor2tensor/data_generators/translate_ende_test.py::TranslateEndeTest::test_vocab_size
tensor2tensor/data_generators/translate_test.py::TranslateTest::testCompileData
tensor2tensor/data_generators/translate_test.py::TranslateTest::test_session
tensor2tensor/data_generators/video_utils_test.py::VideoUtilsTest::testConvertPredictionsToVideoSummariesd5_o6
tensor2tensor/data_generators/video_utils_test.py::VideoUtilsTest::testDecodeInMemoryTrue
tensor2tensor/data_generators/video_utils_test.py::VideoUtilsTest::testVideoAugmentation
tensor2tensor/data_generators/video_utils_test.py::VideoUtilsTest::test_session
tensor2tensor/data_generators/wikisum/utils_test.py::UtilsTest::test_filter_paragraph
tensor2tensor/data_generators/wikisum/utils_test.py::UtilsTest::test_session
tensor2tensor/envs/env_problem_utils_test.py::EnvProblemUtilsTest::test_play_env_problem_randomly
tensor2tensor/envs/env_problem_utils_test.py::EnvProblemUtilsTest::test_play_env_problem_with_policy
tensor2tensor/envs/env_problem_utils_test.py::EnvProblemUtilsTest::test_session
tensor2tensor/envs/gym_env_problem_test.py::GymEnvProblemTest::test_default_processed_rewards_discrete
tensor2tensor/envs/gym_env_problem_test.py::GymEnvProblemTest::test_generate_data
tensor2tensor/envs/gym_env_problem_test.py::GymEnvProblemTest::test_interaction_with_env
tensor2tensor/envs/gym_env_problem_test.py::GymEnvProblemTest::test_per_env_kwargs
tensor2tensor/envs/gym_env_problem_test.py::GymEnvProblemTest::test_problem_dataset_works
tensor2tensor/envs/gym_env_problem_test.py::GymEnvProblemTest::test_resets_properly
tensor2tensor/envs/gym_env_problem_test.py::GymEnvProblemTest::test_reward_range
tensor2tensor/envs/gym_env_problem_test.py::GymEnvProblemTest::test_session
tensor2tensor/envs/gym_env_problem_test.py::GymEnvProblemTest::test_setup
tensor2tensor/envs/gym_spaces_utils_test.py::GymSpacesUtilsTest::test_box_space_encode
tensor2tensor/envs/gym_spaces_utils_test.py::GymSpacesUtilsTest::test_box_space_spec
tensor2tensor/envs/gym_spaces_utils_test.py::GymSpacesUtilsTest::test_discrete_space_encode
tensor2tensor/envs/gym_spaces_utils_test.py::GymSpacesUtilsTest::test_discrete_space_spec
tensor2tensor/envs/gym_spaces_utils_test.py::GymSpacesUtilsTest::test_session
tensor2tensor/envs/mujoco_problems_test.py::ReacherEnvProblemTest::test_registration_and_interaction_with_env_problem
tensor2tensor/envs/mujoco_problems_test.py::ReacherEnvProblemTest::test_session
tensor2tensor/envs/rendered_env_problem_test.py::RenderedEnvProblemTest::test_generate_timesteps
tensor2tensor/envs/rendered_env_problem_test.py::RenderedEnvProblemTest::test_session
tensor2tensor/envs/tic_tac_toe_env_problem_test.py::TicTacToeEnvProblemTest::test_registration_and_interaction_with_env_problem
tensor2tensor/envs/tic_tac_toe_env_problem_test.py::TicTacToeEnvProblemTest::test_session
tensor2tensor/envs/tic_tac_toe_env_test.py::TicTacToeEnvTest::test_env_actions
tensor2tensor/envs/tic_tac_toe_env_test.py::TicTacToeEnvTest::test_keep_playing
tensor2tensor/envs/tic_tac_toe_env_test.py::TicTacToeEnvTest::test_session
tensor2tensor/envs/tic_tac_toe_env_test.py::TicTacToeEnvTest::test_start
tensor2tensor/envs/time_step_test.py::TimeStepTest::test_create_time_step
tensor2tensor/envs/time_step_test.py::TimeStepTest::test_replace
tensor2tensor/envs/time_step_test.py::TimeStepTest::test_session
tensor2tensor/envs/trajectory_test.py::TrajectoryTest::test_add_time_step
tensor2tensor/envs/trajectory_test.py::TrajectoryTest::test_as_numpy
tensor2tensor/envs/trajectory_test.py::TrajectoryTest::test_change_last_time_step
tensor2tensor/envs/trajectory_test.py::TrajectoryTest::test_empty_trajectory
tensor2tensor/envs/trajectory_test.py::TrajectoryTest::test_observation_np
tensor2tensor/envs/trajectory_test.py::TrajectoryTest::test_reward
tensor2tensor/envs/trajectory_test.py::TrajectoryTest::test_session
tensor2tensor/envs/trajectory_test.py::TrajectoryTest::test_truncate_and_last_n_observations_np
tensor2tensor/envs/trajectory_test.py::BatchTrajectoryTest::test_creation
tensor2tensor/envs/trajectory_test.py::BatchTrajectoryTest::test_desired_placement_of_rewards_and_actions
tensor2tensor/envs/trajectory_test.py::BatchTrajectoryTest::test_load_from_directory
tensor2tensor/envs/trajectory_test.py::BatchTrajectoryTest::test_num_time_steps
tensor2tensor/envs/trajectory_test.py::BatchTrajectoryTest::test_observations_np
tensor2tensor/envs/trajectory_test.py::BatchTrajectoryTest::test_parse_trajectory_file_name
tensor2tensor/envs/trajectory_test.py::BatchTrajectoryTest::test_reset_all
tensor2tensor/envs/trajectory_test.py::BatchTrajectoryTest::test_reset_some
tensor2tensor/envs/trajectory_test.py::BatchTrajectoryTest::test_session
tensor2tensor/envs/trajectory_test.py::BatchTrajectoryTest::test_step
tensor2tensor/envs/trajectory_test.py::BatchTrajectoryTest::test_truncate
tensor2tensor/layers/area_attention_test.py::AreaAttentionTest::test1DAreaMax
tensor2tensor/layers/area_attention_test.py::AreaAttentionTest::test2DAreaMax
tensor2tensor/layers/area_attention_test.py::AreaAttentionTest::testAreaMean
tensor2tensor/layers/area_attention_test.py::AreaAttentionTest::testComputeAreaFeatures1D
tensor2tensor/layers/area_attention_test.py::AreaAttentionTest::testComputeAreaFeatures2D
tensor2tensor/layers/area_attention_test.py::AreaAttentionTest::test_session
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::test2dBlockRasterScanMask
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::test2dGather
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::test2dGatherAndScatterInvertibility
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testAddPositionalEmbedding
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testAddPositionalEmbeddingNd0
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testAddPositionalEmbeddingNd1
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testAddPositionalEmbeddingNd2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testAddTimingSignalsGivenPositions
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testAddTimingSignalsGivenPositionsEquivalent
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testAttentionBiasLocal
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testBiasBatchCoordinates
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testBiasFuture
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testBreakIntoBlocksND
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testBreakIntoMemoryBlocksND
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testCausalAttentionBiasND
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testCombineHeadsND0
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testCombineHeadsND1
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testCombineHeadsND2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testContentMaskedLocalAttentionND
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDecodeStepToIndex0
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDecodeStepToIndex1
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDecodeStepToIndex2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDilatedAttention
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDilatedAttentionbatches
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDilatedAttentionblock_length
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDilatedAttentiondynamic_batch
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDotProductAttention
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDotProductAttentionRelative
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDotProductUnMaskedAttentionRelative2d
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDotProductUnMaskedAttentionRelative2dSharedOneRow0
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDotProductUnMaskedAttentionRelative2dSharedOneRow1
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDotProductUnMaskedAttentionRelative2dSharedOneRow2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDotProductUnMaskedAttentionRelative2dSharedOneRow3
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDotProductUnMaskedAttentionRelative2dSharedOneRow4
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDotProductUnMaskedAttentionRelative2dSharedOneRow5
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDotProductUnMaskedAttentionRelativeV2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDotProductUnmaskedAttentionLocal2dTpu
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testDotProductUnmaskedAttentionLocal2dTpuSimple
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testExtractblocks
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testFlattenBlocksND0
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testFlattenBlocksND1
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testFlattenBlocksND2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testGet2dLocalMemory
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testGetItemAtDecodeStep
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testGetLeftRightBlocks
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testGetMemoryRegion
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testGetShiftedCenterBlocks
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testHardenAttentionAllZeroshard_top_k
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testHardenAttentionAllZerossampled_top_k
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testHardenAttentionWeightshard_top_k
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testHardenAttentionWeightssampled_top_k_2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testHardenAttentionWeightssampled_top_k_default
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testLocalUnmaskedAttention1Ddifferent_depth_v
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testLocalUnmaskedAttention1Ddynamic_batch
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testLocalUnmaskedAttention1Dmatching_block_length
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testLocalUnmaskedAttention1Dunmatching_block_length
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testLocalUnmaskedAttention2Ddynamic_batch
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testLocalUnmaskedAttention2Dmatching_block_length
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testLocalUnmaskedAttention2Dunmatching_block_length
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedDilatedAttention
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedDilatedAttentionbatches
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedDilatedAttentionblock_length
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedDilatedAttentiondynamic_batch
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedLocalAttention1D
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedLocalAttention1Dbatches
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedLocalAttention1Dblock_length
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedLocalAttention1Ddepth_v
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedLocalAttention1Ddynamic_batch
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedLocalAttention2D
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedLocalAttention2Dbatches
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedLocalAttention2Ddynamic_batch
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedLocalAttention2Dquery_shape
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedRelativeLocalAttentionV2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedRelativeLocalAttentionV2AddRelativeValues
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedRelativeLocalAttentionV2SeqShorterThanBlockLength
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedRelativeLocalAttentionV2SeqShorterThanTwiceBlockLength
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedWithinBlockLocalAttention1D
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedWithinBlockLocalAttention1Dbatches
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedWithinBlockLocalAttention1Dblock_length
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedWithinBlockLocalAttention1Ddepth_v
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMaskedWithinBlockLocalAttention1Ddynamic_batch
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMultiheadAttentionND0
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMultiheadAttentionND1
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMultiheadAttentionND2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMultiheadAttentionWithLayerCollection
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testMultiheadSelfAttentionMemoryEfficient
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testPadToMultipleND0
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testPadToMultipleND1
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testPadToMultipleND2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testPutBackBlocksND
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testPutItemAtDecodeStep
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testRelativeAttentionV2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testRelativeAttentionV2MaxRelativeLargerThanLength
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testRelativeAttentionV2SharedRel
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testRelativeAttentionV2Unmasked
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testRelativeAttentionV2UnmaskedRelativeLargerThanLength
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testRelativeAttentionV2UnmaskedSharedRel
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testRightShiftBlockwiseND
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testSelectBlockForDecodeStep
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testShapeMaskedLocalAttentionND0
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testShapeMaskedLocalAttentionND1
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testShapeMaskedLocalAttentionND2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testSplitAlongWidth
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testSplitHeadsND0
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testSplitHeadsND1
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testSplitHeadsND2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testUnflattenBlocksND0
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testUnflattenBlocksND1
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::testUnflattenBlocksND2
tensor2tensor/layers/common_attention_test.py::CommonAttentionTest::test_session
tensor2tensor/layers/common_image_attention_test.py::CommonImageAttentionTest::testCreateOutputTrainMode0
tensor2tensor/layers/common_image_attention_test.py::CommonImageAttentionTest::testCreateOutputTrainMode1
tensor2tensor/layers/common_image_attention_test.py::CommonImageAttentionTest::testPostProcessImageInferMode0
tensor2tensor/layers/common_image_attention_test.py::CommonImageAttentionTest::testPostProcessImageInferMode1
tensor2tensor/layers/common_image_attention_test.py::CommonImageAttentionTest::testPostProcessImageTrainMode0
tensor2tensor/layers/common_image_attention_test.py::CommonImageAttentionTest::testPostProcessImageTrainMode1
tensor2tensor/layers/common_image_attention_test.py::CommonImageAttentionTest::testTransformerDecoderLayersGlobal
tensor2tensor/layers/common_image_attention_test.py::CommonImageAttentionTest::test_session
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testAddTimingSignal
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testApplyNormBatch
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testApplyNormLayer
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testApplyNormNoam
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testApplyNormNone
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testBELU
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testBReLU
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testConv
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testConv1d
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testConvBlock
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testConvBlockDownsample
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testConvGRU
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testConvHiddenReluMemoryEfficient
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testConvLSTM
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testConvStride2MultiStep
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testCycleGANUpsampleBilinearUpsampleConv
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testCycleGANUpsampleConv2dTranspose
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testCycleGANUpsampleNnUpsampleConv
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testDeconvStride2MultiStep
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testDenseWithLayerCollection
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testDiscretizedMixLogisticLoss
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testDmlLoss0
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testDmlLoss1
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testDmlLoss2
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testDmlLoss3
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testEmbedding
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testFactoredTensorImplicitConversion
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testFlatten4D3D
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testGetTimingSignal
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testGlobalPool1d
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testGroupNorm
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testIndexLastDimWithIndices
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testLayerNorm
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testLinearSetLayer
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testNAC
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testNALU
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testNALUzeros
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testPadToSameLength
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testPaddingCrossEntropyFactored
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testPaddingCrossEntropyFactoredGrad
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testPool
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testRavanbakhshSetLayer
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testSRU
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testSampleFromDiscretizedMixLogistic
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testSaturatingSigmoid
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testSeparableConv
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testSeparableConvBlock
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testShakeShake
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testShiftLeft
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testSpectralNorm
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testSubSeparableConv
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testSubSeparableConvBlock
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testTopKthIterativeShape
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testTopKthIterativeValue
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testWeightsMultiProblem
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::testWeightsMultiProblemAll
tensor2tensor/layers/common_layers_test.py::CommonLayersTest::test_session
tensor2tensor/layers/common_layers_test.py::FnWithCustomGradTest::testCorrectness
tensor2tensor/layers/common_layers_test.py::FnWithCustomGradTest::testCustomGrad
tensor2tensor/layers/common_layers_test.py::FnWithCustomGradTest::test_session
tensor2tensor/layers/common_layers_test.py::RecomputeTest::testRecompute
tensor2tensor/layers/common_layers_test.py::RecomputeTest::test_session
tensor2tensor/layers/common_video_test.py::CommonVideoTest::testBasicLstm
tensor2tensor/layers/common_video_test.py::CommonVideoTest::testDynamicTileAndConcat
tensor2tensor/layers/common_video_test.py::CommonVideoTest::testExtractRandomVideoPatchdefault
tensor2tensor/layers/common_video_test.py::CommonVideoTest::testExtractRandomVideoPatchten_frames
tensor2tensor/layers/common_video_test.py::CommonVideoTest::testExtractRandomVideoPatchtwo_frames
tensor2tensor/layers/common_video_test.py::CommonVideoTest::testGifSummary
tensor2tensor/layers/common_video_test.py::CommonVideoTest::testScheduledSampleCountEnd
tensor2tensor/layers/common_video_test.py::CommonVideoTest::testScheduledSampleCountMid
tensor2tensor/layers/common_video_test.py::CommonVideoTest::testScheduledSampleCountStart
tensor2tensor/layers/common_video_test.py::CommonVideoTest::testScheduledSampleProbEnd
tensor2tensor/layers/common_video_test.py::CommonVideoTest::testScheduledSampleProbMid
tensor2tensor/layers/common_video_test.py::CommonVideoTest::testScheduledSampleProbStart
tensor2tensor/layers/common_video_test.py::CommonVideoTest::test_session
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testBitToIntOnes
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testBitToIntZeros
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testDiscreteBottleneckVQ
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testDiscreteBottleneckVQCond
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testGetVQBottleneck
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testGumbelSoftmaxDiscreteBottleneck
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testIntToBitOnes
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testIntToBitZeros
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testNearestNeighbors
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testProjectHidden
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testSliceHiddenOnes
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testSliceHiddenZeros
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testVQDiscreteBottleneck
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testVQDiscreteUnbottlenck
tensor2tensor/layers/discretization_test.py::DiscretizationTest::testVQNearestNeighbors
tensor2tensor/layers/discretization_test.py::DiscretizationTest::test_session
tensor2tensor/layers/latent_layers_test.py::LatentLayersTest::testComputeBitsAndNats
tensor2tensor/layers/latent_layers_test.py::LatentLayersTest::testTransformerAutoencoder
tensor2tensor/layers/latent_layers_test.py::LatentLayersTest::test_session
tensor2tensor/layers/modalities_test.py::ModalityTest::testGetForAllModalities
tensor2tensor/layers/modalities_test.py::ModalityTest::testSymbolModalityInputs
tensor2tensor/layers/modalities_test.py::ModalityTest::testSymbolModalityTargets
tensor2tensor/layers/modalities_test.py::ModalityTest::testSymbolModalityTargetsFactored
tensor2tensor/layers/modalities_test.py::ModalityTest::test_session
tensor2tensor/layers/ngram_test.py::NGramTest::testNGramLayerOutput
tensor2tensor/layers/ngram_test.py::NGramTest::testNGramLayerShape
tensor2tensor/layers/ngram_test.py::NGramTest::test_session
tensor2tensor/layers/transformer_memory_test.py::TransformerMemoryTest::testLoss
tensor2tensor/layers/transformer_memory_test.py::TransformerMemoryTest::testRead
tensor2tensor/layers/transformer_memory_test.py::TransformerMemoryTest::testReset
tensor2tensor/layers/transformer_memory_test.py::TransformerMemoryTest::testWrite
tensor2tensor/layers/transformer_memory_test.py::TransformerMemoryTest::test_session
tensor2tensor/metrics/video_conditional_fvd_test.py::VideoConditionalFvdTest::test_sample
tensor2tensor/metrics/video_conditional_fvd_test.py::VideoConditionalFvdTest::test_session
tensor2tensor/models/basic_test.py::BasicTest::testBasicFcRelu
tensor2tensor/models/basic_test.py::BasicTest::test_session
tensor2tensor/models/bytenet_test.py::ByteNetTest::testByteNet
tensor2tensor/models/bytenet_test.py::ByteNetTest::test_session
tensor2tensor/models/evolved_transformer_test.py::EvolvedTransformerTest::testAllWeightsTrainableByDefault
tensor2tensor/models/evolved_transformer_test.py::EvolvedTransformerTest::testBeamVsFast
tensor2tensor/models/evolved_transformer_test.py::EvolvedTransformerTest::testEvolvedTransformer
tensor2tensor/models/evolved_transformer_test.py::EvolvedTransformerTest::testFrozenWeightsUnchangedByTraining
tensor2tensor/models/evolved_transformer_test.py::EvolvedTransformerTest::testGreedyFastTPUVsNonTPU
tensor2tensor/models/evolved_transformer_test.py::EvolvedTransformerTest::testGreedySlowTPUVsNonTPU
tensor2tensor/models/evolved_transformer_test.py::EvolvedTransformerTest::testGreedyTPUSlowVsFast
tensor2tensor/models/evolved_transformer_test.py::EvolvedTransformerTest::testSlowVsFast
tensor2tensor/models/evolved_transformer_test.py::EvolvedTransformerTest::testSlowVsFastNoInput
tensor2tensor/models/evolved_transformer_test.py::EvolvedTransformerTest::test_session
tensor2tensor/models/image_transformer_2d_test.py::Img2imgTransformerTest::testImg2imgTransformer
tensor2tensor/models/image_transformer_2d_test.py::Img2imgTransformerTest::test_session
tensor2tensor/models/image_transformer_2d_test.py::Imagetransformer2dTest::testImagetransformer2d
tensor2tensor/models/image_transformer_2d_test.py::Imagetransformer2dTest::test_session
tensor2tensor/models/image_transformer_test.py::ImagetransformerTest::testImagetransformerImageTransformerCat
tensor2tensor/models/image_transformer_test.py::ImagetransformerTest::testImagetransformerImageTransformerDmol
tensor2tensor/models/image_transformer_test.py::ImagetransformerTest::test_session
tensor2tensor/models/lstm_test.py::LSTMTest::testLSTMSeq2Seq
tensor2tensor/models/lstm_test.py::LSTMTest::testLSTMSeq2SeqAttention
tensor2tensor/models/lstm_test.py::LSTMTest::testLSTMSeq2seqAttentionBidirectionalEncoder
tensor2tensor/models/lstm_test.py::LSTMTest::testLSTMSeq2seqBidirectionalEncoder
tensor2tensor/models/lstm_test.py::LSTMTest::test_session
tensor2tensor/models/mtf_image_transformer_test.py::MtfImageTransformerTest::testMtfImageTransformer
tensor2tensor/models/mtf_image_transformer_test.py::MtfImageTransformerTest::testMtfImageTransformerDataParallel
tensor2tensor/models/mtf_image_transformer_test.py::MtfImageTransformerTest::testMtfImageTransformerModelParallel
tensor2tensor/models/mtf_image_transformer_test.py::MtfImageTransformerTest::test_session
tensor2tensor/models/mtf_transformer_test.py::MtfTransformerTest::testMtfTransformer
tensor2tensor/models/mtf_transformer_test.py::MtfTransformerTest::testMtfTransformerDataModelParallel
tensor2tensor/models/mtf_transformer_test.py::MtfTransformerTest::testMtfTransformerDataParallel
tensor2tensor/models/mtf_transformer_test.py::MtfTransformerTest::testMtfTransformerEncoderDataModelParallel
tensor2tensor/models/mtf_transformer_test.py::MtfTransformerTest::testMtfTransformerModelParallel
tensor2tensor/models/mtf_transformer_test.py::MtfTransformerTest::test_session
tensor2tensor/models/neural_gpu_test.py::NeuralGPUTest::testNeuralGPU
tensor2tensor/models/neural_gpu_test.py::NeuralGPUTest::test_session
tensor2tensor/models/resnet_test.py::ResnetTest::testResnetLarge
tensor2tensor/models/resnet_test.py::ResnetTest::test_session
tensor2tensor/models/revnet_test.py::RevnetTest::testEndToEnd
tensor2tensor/models/revnet_test.py::RevnetTest::testEndToEnd3D
tensor2tensor/models/revnet_test.py::RevnetTest::testFinalBlock
tensor2tensor/models/revnet_test.py::RevnetTest::testFinalBlock3D
tensor2tensor/models/revnet_test.py::RevnetTest::testH
tensor2tensor/models/revnet_test.py::RevnetTest::testHStride
tensor2tensor/models/revnet_test.py::RevnetTest::testInit
tensor2tensor/models/revnet_test.py::RevnetTest::testInit3D
tensor2tensor/models/revnet_test.py::RevnetTest::testUnit1
tensor2tensor/models/revnet_test.py::RevnetTest::testUnit2
tensor2tensor/models/revnet_test.py::RevnetTest::testUnit3
tensor2tensor/models/revnet_test.py::RevnetTest::testUnit3D
tensor2tensor/models/revnet_test.py::RevnetTest::testUnit4
tensor2tensor/models/revnet_test.py::RevnetTest::test_session
tensor2tensor/models/slicenet_test.py::SliceNetTest::testSliceNet
tensor2tensor/models/slicenet_test.py::SliceNetTest::testSliceNetImageToText
tensor2tensor/models/slicenet_test.py::SliceNetTest::test_session
tensor2tensor/models/transformer_test.py::TransformerTest::testBeamDecodeWithRelativeAttention
tensor2tensor/models/transformer_test.py::TransformerTest::testBeamVsFast
tensor2tensor/models/transformer_test.py::TransformerTest::testGreedyFastTPUVsNonTPU
tensor2tensor/models/transformer_test.py::TransformerTest::testGreedySlowTPUVsNonTPU
tensor2tensor/models/transformer_test.py::TransformerTest::testGreedyTPUSlowVsFast
tensor2tensor/models/transformer_test.py::TransformerTest::testLibrispeechMultihead
tensor2tensor/models/transformer_test.py::TransformerTest::testLibrispeechSlowVsFast
tensor2tensor/models/transformer_test.py::TransformerTest::testLibrispeechWithAreaAttention
tensor2tensor/models/transformer_test.py::TransformerTest::testSlowVsFast
tensor2tensor/models/transformer_test.py::TransformerTest::testSlowVsFastNoInput
tensor2tensor/models/transformer_test.py::TransformerTest::testTransformer
tensor2tensor/models/transformer_test.py::TransformerTest::testTransformerLibrispeech
tensor2tensor/models/transformer_test.py::TransformerTest::testTransformerRelative
tensor2tensor/models/transformer_test.py::TransformerTest::testTransformerWithEncoderDecoderAttentionLoss
tensor2tensor/models/transformer_test.py::TransformerTest::testTransformerWithoutProblem
tensor2tensor/models/transformer_test.py::TransformerTest::test_session
tensor2tensor/models/transformer_test.py::TransformerScorerTest::testReturnsScores
tensor2tensor/models/transformer_test.py::TransformerScorerTest::testVarNames
tensor2tensor/models/transformer_test.py::TransformerScorerTest::test_session
tensor2tensor/models/xception_test.py::XceptionTest::testXceptionLargeImage
tensor2tensor/models/xception_test.py::XceptionTest::testXceptionSmallImage
tensor2tensor/models/xception_test.py::XceptionTest::test_session
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer0
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer1
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer10
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer100
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer101
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer11
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer12
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer13
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer14
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer15
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer16
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer17
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer18
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer19
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer2
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer20
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer21
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer22
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer23
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer24
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer25
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer26
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer27
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer28
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer29
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer3
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer30
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer31
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer32
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer33
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer34
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer35
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer36
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer37
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer38
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer39
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer4
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer40
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer41
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer42
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer43
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer44
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer45
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer46
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer47
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer48
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer49
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer5
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer50
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer51
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer52
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer53
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer54
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer55
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer56
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer57
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer58
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer59
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer6
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer60
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer61
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer62
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer63
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer64
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer65
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer66
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer67
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer68
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer69
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer7
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer70
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer71
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer72
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer73
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer74
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer75
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer76
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer77
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer78
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer79
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer8
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer80
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer81
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer82
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer83
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer84
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer85
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer86
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer87
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer88
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer89
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer9
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer90
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer91
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer92
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer93
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer94
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer95
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer96
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer97
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer98
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_layer99
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_decoder_registry
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer0
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer1
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer10
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer11
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer12
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer13
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer14
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer15
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer16
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer17
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer18
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer19
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer2
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer20
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer21
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer22
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer23
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer24
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer25
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer26
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer27
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer28
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer29
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer3
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer30
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer31
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer32
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer33
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer34
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer35
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer36
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer37
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer38
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer39
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer4
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer40
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer41
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer42
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer43
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer44
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer45
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer46
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer47
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer48
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer49
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer5
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer50
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer51
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer52
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer53
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer54
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer55
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer56
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer57
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer58
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer59
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer6
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer60
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer61
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer62
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer63
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer64
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer65
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer66
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer67
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer68
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer69
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer7
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer70
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer71
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer72
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer73
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer74
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer75
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer76
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer77
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer78
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer79
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer8
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer80
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer81
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer82
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer83
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer84
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer85
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer86
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer87
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer88
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer89
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer9
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer90
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer91
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer92
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer93
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer94
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer95
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer96
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer97
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_layer98
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_encoder_registry
tensor2tensor/models/neural_architecture_search/nas_layers_test.py::LayersTest::test_session
tensor2tensor/models/neural_architecture_search/nas_model_test.py::NasSeq2SeqTest::test_calculate_branching_model_parameters_decoder_resize0
tensor2tensor/models/neural_architecture_search/nas_model_test.py::NasSeq2SeqTest::test_calculate_branching_model_parameters_decoder_resize1
tensor2tensor/models/neural_architecture_search/nas_model_test.py::NasSeq2SeqTest::test_calculate_branching_model_parameters_output_size_last_two
tensor2tensor/models/neural_architecture_search/nas_model_test.py::NasSeq2SeqTest::test_calculate_branching_model_parameters_output_size_only_final
tensor2tensor/models/neural_architecture_search/nas_model_test.py::NasSeq2SeqTest::test_calculate_branching_model_parameters_transformer0
tensor2tensor/models/neural_architecture_search/nas_model_test.py::NasSeq2SeqTest::test_calculate_branching_model_parameters_transformer1
tensor2tensor/models/neural_architecture_search/nas_model_test.py::NasSeq2SeqTest::test_nas_decoder_resizing_output
tensor2tensor/models/neural_architecture_search/nas_model_test.py::NasSeq2SeqTest::test_nas_seq2seq
tensor2tensor/models/neural_architecture_search/nas_model_test.py::NasSeq2SeqTest::test_session
tensor2tensor/models/research/autoencoders_test.py::AutoencoderTest::testAutoencoderAutoregressive
tensor2tensor/models/research/autoencoders_test.py::AutoencoderTest::testAutoencoderBasic
tensor2tensor/models/research/autoencoders_test.py::AutoencoderTest::testAutoencoderBasicDiscrete
tensor2tensor/models/research/autoencoders_test.py::AutoencoderTest::testAutoencoderOrderedDiscrete
tensor2tensor/models/research/autoencoders_test.py::AutoencoderTest::testAutoencoderOrderedDiscreteVQ
tensor2tensor/models/research/autoencoders_test.py::AutoencoderTest::testAutoencoderResidual
tensor2tensor/models/research/autoencoders_test.py::AutoencoderTest::testAutoencoderResidualDiscrete
tensor2tensor/models/research/autoencoders_test.py::AutoencoderTest::test_session
tensor2tensor/models/research/gene_expression_test.py::GeneExpressionModelsTest::testGeneExpressionModels
tensor2tensor/models/research/gene_expression_test.py::GeneExpressionModelsTest::test_session
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_actnorm
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_actnorm_3d
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_add_edge_bias
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_conv2d
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_conv3d
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_conv_stack_gatu_act
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_conv_stack_relu_act
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_encoder_decoder
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_encoder_decoder_practical_usage
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_get_variable_ddi
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_invertibility_actnorm
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_invertibility_additive
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_invertibility_additive_drop
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_invertibility_affine
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_invertibility_affine_drop
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_invertibility_inv_1x1
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_latent_dist_encoder_conv3d_no_skip
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_latent_dist_encoder_conv3d_no_skip_drop
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_latent_dist_encoder_conv3d_no_skip_drop_noise
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_latent_dist_encoder_conv3d_skip
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_latent_dist_encoder_conv3d_skip_drop
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_latent_dist_encoder_conv_net_no_skip
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_latent_dist_encoder_conv_net_skip
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_latent_dist_encoder_lstm_no_skip
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_latent_dist_encoder_lstm_skip
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_latent_to_dist
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_revnet_reversibility_add_rev_step
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_revnet_reversibility_add_revnet
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_revnet_reversibility_aff_rev_step
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_revnet_reversibility_aff_revnet
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_scale_gaussian_prior
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_session
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_split
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_split_latent_conditioning
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temperature_normal_temp_0.0
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temperature_normal_temp_0.1
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temperature_normal_temp_0.3
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temperature_normal_temp_0.7
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temperature_normal_temp_0.9
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temperature_normal_temp_1.0
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temporal_latent_to_dist_dil_gatu
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temporal_latent_to_dist_dil_gatu_drop
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temporal_latent_to_dist_dil_gatu_drop_noise
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temporal_latent_to_dist_dil_gatu_drop_single_step
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temporal_latent_to_dist_dil_relu
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temporal_latent_to_dist_dil_relu_drop
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temporal_latent_to_dist_gatu_drop_single_step
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temporal_latent_to_dist_no_dil_gatu
tensor2tensor/models/research/glow_ops_test.py::GlowOpsTest::test_temporal_latent_to_dist_no_dil_relu
tensor2tensor/models/research/glow_test.py::GlowModelTest::test_glow
tensor2tensor/models/research/glow_test.py::GlowModelTest::test_glow_inference
tensor2tensor/models/research/glow_test.py::GlowModelTest::test_session
tensor2tensor/models/research/neural_stack_test.py::NeuralStackCellTest::test_cell_shapes
tensor2tensor/models/research/neural_stack_test.py::NeuralStackCellTest::test_push_pop
tensor2tensor/models/research/neural_stack_test.py::NeuralStackCellTest::test_session
tensor2tensor/models/research/neural_stack_test.py::NeuralQueueCellTest::test_enqueue_dequeue
tensor2tensor/models/research/neural_stack_test.py::NeuralQueueCellTest::test_session
tensor2tensor/models/research/neural_stack_test.py::NeuralDequeCellTest::test_cell_shapes
tensor2tensor/models/research/neural_stack_test.py::NeuralDequeCellTest::test_enqueue_dequeue
tensor2tensor/models/research/neural_stack_test.py::NeuralDequeCellTest::test_session
tensor2tensor/models/research/neural_stack_test.py::NeuralStackModelTest::test_model_shapes
tensor2tensor/models/research/neural_stack_test.py::NeuralStackModelTest::test_session
tensor2tensor/models/research/transformer_aux_test.py::TransformerAuxTest::test_session
tensor2tensor/models/research/transformer_aux_test.py::TransformerAuxTest::test_shift_and_pad0
tensor2tensor/models/research/transformer_aux_test.py::TransformerAuxTest::test_shift_and_pad1
tensor2tensor/models/research/transformer_aux_test.py::TransformerAuxTest::test_shift_and_pad2
tensor2tensor/models/research/transformer_aux_test.py::TransformerAuxTest::test_shift_and_pad3
tensor2tensor/models/research/transformer_aux_test.py::TransformerAuxTest::test_transformer_aux_body
tensor2tensor/models/research/transformer_revnet_test.py::TransformerRevnetTest::testTransformer
tensor2tensor/models/research/transformer_revnet_test.py::TransformerRevnetTest::test_session
tensor2tensor/models/research/transformer_vae_test.py::TransformerVaeTest::testTransformerAEOnDVQ
tensor2tensor/models/research/transformer_vae_test.py::TransformerVaeTest::test_session
tensor2tensor/models/research/universal_transformer_test.py::UniversalTransformerTest::testTransformer
tensor2tensor/models/research/universal_transformer_test.py::UniversalTransformerTest::test_session
tensor2tensor/models/research/vqa_attention_test.py::VqaAttentionBaselineTest::testVqaAttentionBaseline
tensor2tensor/models/research/vqa_attention_test.py::VqaAttentionBaselineTest::test_session
tensor2tensor/models/video/basic_deterministic_test.py::NextFrameTest::testBasicDeterministic
tensor2tensor/models/video/basic_deterministic_test.py::NextFrameTest::test_session
tensor2tensor/models/video/basic_recurrent_test.py::NextFrameTest::testBasicDeterministic
tensor2tensor/models/video/basic_recurrent_test.py::NextFrameTest::test_session
tensor2tensor/models/video/basic_stochastic_test.py::NextFrameTest::testBasicStochastic
tensor2tensor/models/video/basic_stochastic_test.py::NextFrameTest::test_session
tensor2tensor/models/video/emily_test.py::NextFrameTest::testEmily
tensor2tensor/models/video/emily_test.py::NextFrameTest::test_session
tensor2tensor/models/video/nfg_conv3d_test.py::NextFrameGlowConv3DTest::testGlowTrainAndDecodeconv3d_dil
tensor2tensor/models/video/nfg_conv3d_test.py::NextFrameGlowConv3DTest::testGlowTrainAndDecodeconv3d_net
tensor2tensor/models/video/nfg_conv3d_test.py::NextFrameGlowConv3DTest::testGlowTrainAndDecodeconv3d_net_gatu
tensor2tensor/models/video/nfg_conv3d_test.py::NextFrameGlowConv3DTest::test_session
tensor2tensor/models/video/nfg_conv_lstm_test.py::NextFrameGlowConv3DTest::testGlowTrainAndDecodein_3_out_2_lstm
tensor2tensor/models/video/nfg_conv_lstm_test.py::NextFrameGlowConv3DTest::testGlowTrainAndDecodelstm_pretrain
tensor2tensor/models/video/nfg_conv_lstm_test.py::NextFrameGlowConv3DTest::test_session
tensor2tensor/models/video/nfg_conv_test.py::NextFrameGlowConvTest::testGlowTrainAndDecodeconv_net_cond_first
tensor2tensor/models/video/nfg_conv_test.py::NextFrameGlowConvTest::testGlowTrainAndDecodein_3_out_2_conv
tensor2tensor/models/video/nfg_conv_test.py::NextFrameGlowConvTest::test_session
tensor2tensor/models/video/nfg_uncond_test.py::NfgUncondTest::testGlowTrainAndDecodein_1_out_1
tensor2tensor/models/video/nfg_uncond_test.py::NfgUncondTest::testGlowTrainAndDecodeuncond
tensor2tensor/models/video/nfg_uncond_test.py::NfgUncondTest::test_session
tensor2tensor/models/video/savp_test.py::NextFrameTest::testInvalidVAEGANCombinations
tensor2tensor/models/video/savp_test.py::NextFrameTest::testSavpGAN
tensor2tensor/models/video/savp_test.py::NextFrameTest::testSavpGANVAE
tensor2tensor/models/video/savp_test.py::NextFrameTest::testSavpVAE
tensor2tensor/models/video/savp_test.py::NextFrameTest::test_session
tensor2tensor/models/video/sv2p_test.py::NextFrameTest::testSv2p
tensor2tensor/models/video/sv2p_test.py::NextFrameTest::testSv2pTwoFrames
tensor2tensor/models/video/sv2p_test.py::NextFrameTest::testSv2pWithActions
tensor2tensor/models/video/sv2p_test.py::NextFrameTest::testSv2pWithActionsAndRewards
tensor2tensor/models/video/sv2p_test.py::NextFrameTest::testSv2pWithActionsAndRewardsExternalLoss
tensor2tensor/models/video/sv2p_test.py::NextFrameTest::test_session
tensor2tensor/rl/batch_dqn_agent_test.py::BatchDQNAgentTest::testBeginEpisode
tensor2tensor/rl/batch_dqn_agent_test.py::BatchDQNAgentTest::testCreateAgentWithDefaults
tensor2tensor/rl/batch_dqn_agent_test.py::BatchDQNAgentTest::test_session
tensor2tensor/rl/batch_runner_test.py::BatchedRunnerTest::testLogExperiment
tensor2tensor/rl/batch_runner_test.py::BatchedRunnerTest::testRunEpisodeBatch
tensor2tensor/rl/batch_runner_test.py::BatchedRunnerTest::testRunOneEpisodeWithLowMaxSteps
tensor2tensor/rl/batch_runner_test.py::BatchedRunnerTest::testRunOneIteration
tensor2tensor/rl/batch_runner_test.py::BatchedRunnerTest::testRunOnePhase
tensor2tensor/rl/batch_runner_test.py::BatchedRunnerTest::test_session
tensor2tensor/rl/gym_utils_test.py::GymUtilsTest::test_gym_registration
tensor2tensor/rl/gym_utils_test.py::GymUtilsTest::test_gym_registration_with_kwargs
tensor2tensor/rl/gym_utils_test.py::GymUtilsTest::test_making_simple_env
tensor2tensor/rl/gym_utils_test.py::GymUtilsTest::test_making_timewrapped_env
tensor2tensor/rl/gym_utils_test.py::GymUtilsTest::test_rendered_env
tensor2tensor/rl/gym_utils_test.py::GymUtilsTest::test_session
tensor2tensor/rl/gym_utils_test.py::GymUtilsTest::test_unlimited_env
tensor2tensor/rl/restarter_test.py::RestarterTest::test_restarts_after_interruption
tensor2tensor/rl/restarter_test.py::RestarterTest::test_runs_in_single_mode
tensor2tensor/rl/restarter_test.py::RestarterTest::test_runs_in_two_modes
tensor2tensor/rl/restarter_test.py::RestarterTest::test_session
tensor2tensor/rl/restarter_test.py::RestarterTest::test_skips_already_done
tensor2tensor/rl/trainer_model_based_recurrent_test.py::ModelRLExperimentRecurrentTest::test_basic_recurrent
tensor2tensor/rl/trainer_model_based_recurrent_test.py::ModelRLExperimentRecurrentTest::test_session
tensor2tensor/rl/trainer_model_based_stochastic_test.py::ModelRLExperimentStochasticTest::test_basic_stochastic
tensor2tensor/rl/trainer_model_based_stochastic_test.py::ModelRLExperimentStochasticTest::test_session
tensor2tensor/rl/trainer_model_based_sv2p_test.py::ModelRLExperimentSv2pTest::test_session
tensor2tensor/rl/trainer_model_based_sv2p_test.py::ModelRLExperimentSv2pTest::test_sv2p
tensor2tensor/rl/trainer_model_based_test.py::ModelRLExperimentTest::test_basic
tensor2tensor/rl/trainer_model_based_test.py::ModelRLExperimentTest::test_session
tensor2tensor/rl/trainer_model_free_test.py::TrainTest::test_session
tensor2tensor/rl/trainer_model_free_test.py::TrainTest::test_train_pong
tensor2tensor/rl/trainer_model_free_test.py::TrainTest::test_train_pong_dqn
tensor2tensor/rl/trainer_model_free_tictactoe_test.py::TrainerModelFreeTicTacToeTest::test_session
tensor2tensor/rl/trainer_model_free_tictactoe_test.py::TrainerModelFreeTicTacToeTest::test_train_tictactoe
tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testComputeTopkScoresAndSeq
tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testGreedyBatchOne
tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testGreedyWithCornerCase
tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testNotGreedyBatchTwoBeamTwoWithAlpha
tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testNotGreedyBeamTwoWithAlpha
tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testNotGreedyBeamTwoWithStopEarly
tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testNotGreedyBeamTwoWithoutStopEarly
tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testShapes
tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testStateBeamTwo
tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testStates
tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testStatesAfterLoop
tensor2tensor/utils/beam_search_test.py::BeamSearchTest::testTPUBeam
tensor2tensor/utils/beam_search_test.py::BeamSearchTest::test_session
tensor2tensor/utils/bleu_hook_test.py::BleuHookTest::testBleuTokenize
tensor2tensor/utils/bleu_hook_test.py::BleuHookTest::testBleuWrapper
tensor2tensor/utils/bleu_hook_test.py::BleuHookTest::testBleuWrapperWithUnicodeLineSeparator
tensor2tensor/utils/bleu_hook_test.py::BleuHookTest::testComputeBleuEqual
tensor2tensor/utils/bleu_hook_test.py::BleuHookTest::testComputeMultipleBatch
tensor2tensor/utils/bleu_hook_test.py::BleuHookTest::testComputeMultipleNgrams
tensor2tensor/utils/bleu_hook_test.py::BleuHookTest::testComputeNotEqual
tensor2tensor/utils/bleu_hook_test.py::BleuHookTest::test_session
tensor2tensor/utils/checkpoint_compatibility_test.py::CheckpointCompatibilityTest::testCompatibility
tensor2tensor/utils/checkpoint_compatibility_test.py::CheckpointCompatibilityTest::test_session
tensor2tensor/utils/data_reader_test.py::DataReaderTest::testBasicExampleReading
tensor2tensor/utils/data_reader_test.py::DataReaderTest::testBatchingSchemeBuckets
tensor2tensor/utils/data_reader_test.py::DataReaderTest::testBatchingSchemeMaxLength
tensor2tensor/utils/data_reader_test.py::DataReaderTest::testLengthFilter
tensor2tensor/utils/data_reader_test.py::DataReaderTest::testPreprocess
tensor2tensor/utils/data_reader_test.py::DataReaderTest::test_session
tensor2tensor/utils/diet_test.py::DietVarTest::testDiet
tensor2tensor/utils/diet_test.py::DietVarTest::test_session
tensor2tensor/utils/expert_utils_test.py::ExpertUtilsTest::testPadRemover
tensor2tensor/utils/expert_utils_test.py::ExpertUtilsTest::testTruncatingDispatcher
tensor2tensor/utils/expert_utils_test.py::ExpertUtilsTest::test_session
tensor2tensor/utils/hparam_test.py::HParamsTest::testBoolParsing
tensor2tensor/utils/hparam_test.py::HParamsTest::testBoolParsingFail
tensor2tensor/utils/hparam_test.py::HParamsTest::testContains
tensor2tensor/utils/hparam_test.py::HParamsTest::testDel
tensor2tensor/utils/hparam_test.py::HParamsTest::testEmpty
tensor2tensor/utils/hparam_test.py::HParamsTest::testFunction
tensor2tensor/utils/hparam_test.py::HParamsTest::testGet
tensor2tensor/utils/hparam_test.py::HParamsTest::testJson
tensor2tensor/utils/hparam_test.py::HParamsTest::testLists
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithBadIndexAssigment1
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithBadIndexAssigment1_IgnoreUnknown
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithBadIndexAssigment2
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithBadIndexAssigment2_IgnoreUnknown
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithBadIndexAssigment3
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithBadIndexAssigment3_IgnoreUnknown
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithIndexAssigment1
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithIndexAssigment1_IgnoreUnknown
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithIndexAssigment2
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithIndexAssigment2_IgnoreUnknown
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithIndexAssigment3
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithIndexAssigment3_IgnoreUnknown
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithIndexAssigment4
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithIndexAssigment4_IgnoreUnknown
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithIndexAssigment5
tensor2tensor/utils/hparam_test.py::HParamsTest::testParseValuesWithIndexAssigment5_IgnoreUnknown
tensor2tensor/utils/hparam_test.py::HParamsTest::testSetFromMap
tensor2tensor/utils/hparam_test.py::HParamsTest::testSetHParam
tensor2tensor/utils/hparam_test.py::HParamsTest::testSetHParamListNonListMismatch
tensor2tensor/utils/hparam_test.py::HParamsTest::testSetHParamTypeMismatch
tensor2tensor/utils/hparam_test.py::HParamsTest::testSomeValues
tensor2tensor/utils/hparam_test.py::HParamsTest::testWithPeriodInVariableName
tensor2tensor/utils/hparam_test.py::HParamsTest::testWithReusedVariables
tensor2tensor/utils/hparam_test.py::HParamsTest::test_session
tensor2tensor/utils/hparams_lib_test.py::HparamsLibTest::testCreateHparamsFromJson
tensor2tensor/utils/hparams_lib_test.py::HparamsLibTest::test_session
tensor2tensor/utils/metrics_hook_test.py::MetricsHookTest::testEarlyStoppingHook
tensor2tensor/utils/metrics_hook_test.py::MetricsHookTest::testPlateauOpHook
tensor2tensor/utils/metrics_hook_test.py::MetricsHookTest::testStop
tensor2tensor/utils/metrics_hook_test.py::MetricsHookTest::test_session
tensor2tensor/utils/metrics_test.py::MetricsTest::testAccuracyMetric
tensor2tensor/utils/metrics_test.py::MetricsTest::testAccuracyTopKMetric
tensor2tensor/utils/metrics_test.py::MetricsTest::testMultilabelMatch3
tensor2tensor/utils/metrics_test.py::MetricsTest::testNegativeLogPerplexity
tensor2tensor/utils/metrics_test.py::MetricsTest::testNegativeLogPerplexityMasked
tensor2tensor/utils/metrics_test.py::MetricsTest::testNegativeLogPerplexityMaskedAssert
tensor2tensor/utils/metrics_test.py::MetricsTest::testPearsonCorrelationCoefficient
tensor2tensor/utils/metrics_test.py::MetricsTest::testPrefixAccuracy
tensor2tensor/utils/metrics_test.py::MetricsTest::testRMSEMetric
tensor2tensor/utils/metrics_test.py::MetricsTest::testRocAuc
tensor2tensor/utils/metrics_test.py::MetricsTest::testSequenceAccuracyMetric
tensor2tensor/utils/metrics_test.py::MetricsTest::testSequenceEditDistanceMetric
tensor2tensor/utils/metrics_test.py::MetricsTest::testSigmoidAccuracy
tensor2tensor/utils/metrics_test.py::MetricsTest::testSigmoidAccuracyOneHot
tensor2tensor/utils/metrics_test.py::MetricsTest::testSigmoidCrossEntropyOneHot
tensor2tensor/utils/metrics_test.py::MetricsTest::testSigmoidPrecisionOneHot
tensor2tensor/utils/metrics_test.py::MetricsTest::testSigmoidRecallOneHot
tensor2tensor/utils/metrics_test.py::MetricsTest::testTwoClassAccuracyMetric
tensor2tensor/utils/metrics_test.py::MetricsTest::testTwoClassLogLikelihood
tensor2tensor/utils/metrics_test.py::MetricsTest::testTwoClassLogLikelihoodVersusOldImplementation
tensor2tensor/utils/metrics_test.py::MetricsTest::testUnpaddedRMSEMetric
tensor2tensor/utils/metrics_test.py::MetricsTest::testWordErrorRateMetric
tensor2tensor/utils/metrics_test.py::MetricsTest::test_session
tensor2tensor/utils/misc_utils_test.py::MiscUtilsTest::test_camelcase_to_snakecase
tensor2tensor/utils/misc_utils_test.py::MiscUtilsTest::test_pprint_hparams
tensor2tensor/utils/misc_utils_test.py::MiscUtilsTest::test_session
tensor2tensor/utils/misc_utils_test.py::MiscUtilsTest::test_snakecase_to_camelcase
tensor2tensor/utils/multistep_optimizer_test.py::MultistepAdamOptimizerTest::testMultistep
tensor2tensor/utils/multistep_optimizer_test.py::MultistepAdamOptimizerTest::testResourceVariables
tensor2tensor/utils/multistep_optimizer_test.py::MultistepAdamOptimizerTest::test_session
tensor2tensor/utils/optimize_test.py::OptimizeTest::test_names0
tensor2tensor/utils/optimize_test.py::OptimizeTest::test_names1
tensor2tensor/utils/optimize_test.py::OptimizeTest::test_names2
tensor2tensor/utils/optimize_test.py::OptimizeTest::test_names3
tensor2tensor/utils/optimize_test.py::OptimizeTest::test_names4
tensor2tensor/utils/optimize_test.py::OptimizeTest::test_names5
tensor2tensor/utils/optimize_test.py::OptimizeTest::test_names6
tensor2tensor/utils/optimize_test.py::OptimizeTest::test_names7
tensor2tensor/utils/optimize_test.py::OptimizeTest::test_names8
tensor2tensor/utils/optimize_test.py::OptimizeTest::test_names9
tensor2tensor/utils/optimize_test.py::OptimizeTest::test_session
tensor2tensor/utils/registry_test.py::RegistryClassTest::testDefaultKeyFn
tensor2tensor/utils/registry_test.py::RegistryClassTest::testGet
tensor2tensor/utils/registry_test.py::RegistryClassTest::testGetterSetter
tensor2tensor/utils/registry_test.py::RegistryClassTest::testIteration
tensor2tensor/utils/registry_test.py::RegistryClassTest::testLen
tensor2tensor/utils/registry_test.py::RegistryClassTest::testMembership
tensor2tensor/utils/registry_test.py::RegistryClassTest::testNoKeyProvided
tensor2tensor/utils/registry_test.py::RegistryClassTest::testTransformer
tensor2tensor/utils/registry_test.py::RegistryClassTest::test_session
tensor2tensor/utils/registry_test.py::EnvProblemRegistryTest::testEnvProblem
tensor2tensor/utils/registry_test.py::EnvProblemRegistryTest::test_session
tensor2tensor/utils/registry_test.py::ModelRegistryTest::testDuplicateRegistration
tensor2tensor/utils/registry_test.py::ModelRegistryTest::testListModels
tensor2tensor/utils/registry_test.py::ModelRegistryTest::testNamedRegistration
tensor2tensor/utils/registry_test.py::ModelRegistryTest::testNonT2TModelRegistration
tensor2tensor/utils/registry_test.py::ModelRegistryTest::testT2TModelRegistration
tensor2tensor/utils/registry_test.py::ModelRegistryTest::testUnknownModel
tensor2tensor/utils/registry_test.py::ModelRegistryTest::test_session
tensor2tensor/utils/registry_test.py::OptimizerRegistryTest::testArgErrorCheck
tensor2tensor/utils/registry_test.py::OptimizerRegistryTest::testGetterSetterInterface
tensor2tensor/utils/registry_test.py::OptimizerRegistryTest::testMembership
tensor2tensor/utils/registry_test.py::OptimizerRegistryTest::testMultipleRegistration
tensor2tensor/utils/registry_test.py::OptimizerRegistryTest::testRegistration
tensor2tensor/utils/registry_test.py::OptimizerRegistryTest::testUnknownOptimizer
tensor2tensor/utils/registry_test.py::OptimizerRegistryTest::test_session
tensor2tensor/utils/registry_test.py::HParamRegistryTest::testDuplicateRegistration
tensor2tensor/utils/registry_test.py::HParamRegistryTest::testHParamSet
tensor2tensor/utils/registry_test.py::HParamRegistryTest::testListHparams
tensor2tensor/utils/registry_test.py::HParamRegistryTest::testNamedRegistration
tensor2tensor/utils/registry_test.py::HParamRegistryTest::testNoneHparams
tensor2tensor/utils/registry_test.py::HParamRegistryTest::testRangeSignatureCheck
tensor2tensor/utils/registry_test.py::HParamRegistryTest::testUnknownHparams
tensor2tensor/utils/registry_test.py::HParamRegistryTest::test_session
tensor2tensor/utils/registry_test.py::RegistryHelpTest::testRegistryHelp
tensor2tensor/utils/registry_test.py::RegistryHelpTest::test_session
tensor2tensor/utils/rouge_test.py::TestRouge2Metric::testRouge2Disjoint
tensor2tensor/utils/rouge_test.py::TestRouge2Metric::testRouge2Identical
tensor2tensor/utils/rouge_test.py::TestRouge2Metric::testRouge2PartialOverlap
tensor2tensor/utils/rouge_test.py::TestRouge2Metric::test_session
tensor2tensor/utils/rouge_test.py::TestRougeLMetric::testRougeLDisjoint
tensor2tensor/utils/rouge_test.py::TestRougeLMetric::testRougeLIdentical
tensor2tensor/utils/rouge_test.py::TestRougeLMetric::testRougeLPartialOverlap
tensor2tensor/utils/rouge_test.py::TestRougeLMetric::test_session
tensor2tensor/utils/rouge_test.py::TestRougeMetricsE2E::testRouge2MetricE2E
tensor2tensor/utils/rouge_test.py::TestRougeMetricsE2E::testRougeLMetricE2E
tensor2tensor/utils/rouge_test.py::TestRougeMetricsE2E::test_session
tensor2tensor/utils/sari_hook_test.py::SariHookTest::testDeletionScore
tensor2tensor/utils/sari_hook_test.py::SariHookTest::testIdsWithZeros
tensor2tensor/utils/sari_hook_test.py::SariHookTest::testKeepScore
tensor2tensor/utils/sari_hook_test.py::SariHookTest::testMatchingOutputAndReference
tensor2tensor/utils/sari_hook_test.py::SariHookTest::testMatchingSentences
tensor2tensor/utils/sari_hook_test.py::SariHookTest::testMatchingSentencesWithRepetitions
tensor2tensor/utils/sari_hook_test.py::SariHookTest::testSariScoreE2E
tensor2tensor/utils/sari_hook_test.py::SariHookTest::testSariSent1
tensor2tensor/utils/sari_hook_test.py::SariHookTest::testSariSent2
tensor2tensor/utils/sari_hook_test.py::SariHookTest::testSariSent3
tensor2tensor/utils/sari_hook_test.py::SariHookTest::test_session
tensor2tensor/utils/t2t_model_test.py::T2TModelTest::testLossSingleWeights
tensor2tensor/utils/t2t_model_test.py::T2TModelTest::testSummarizeLosses
tensor2tensor/utils/t2t_model_test.py::T2TModelTest::test_session
tensor2tensor/utils/test_utils.py::test_main
tensor2tensor/utils/test_utils_test.py::RunInGraphAndEagerTest::test_run_in_graph_and_eager_modes
tensor2tensor/utils/test_utils_test.py::RunInGraphAndEagerTest::test_run_in_graph_and_eager_modes_setup_in_same_mode
tensor2tensor/utils/test_utils_test.py::RunInGraphAndEagerTest::test_session
tensor2tensor/utils/trainer_lib_test.py::TrainerLibTest::testCreateHparams
tensor2tensor/utils/trainer_lib_test.py::TrainerLibTest::testExperiment
tensor2tensor/utils/trainer_lib_test.py::TrainerLibTest::testExperimentWithClass
tensor2tensor/utils/trainer_lib_test.py::TrainerLibTest::testModel
tensor2tensor/utils/trainer_lib_test.py::TrainerLibTest::testMultipleTargetModalities
tensor2tensor/utils/trainer_lib_test.py::TrainerLibTest::test_session
tensor2tensor/utils/video_metrics_test.py::VideoMetricsTest::test_reduce_to_best_decode
tensor2tensor/utils/video_metrics_test.py::VideoMetricsTest::test_session
tensor2tensor/utils/yellowfin_test.py::YellowFinTest::testLrMu
tensor2tensor/utils/yellowfin_test.py::YellowFinTest::testMeasurement
tensor2tensor/utils/yellowfin_test.py::YellowFinTest::test_session
tensor2tensor/visualization/visualization_test.py::VisualizationTest::test_build_model_beam
tensor2tensor/visualization/visualization_test.py::VisualizationTest::test_build_model_greedy
tensor2tensor/visualization/visualization_test.py::VisualizationTest::test_get_vis_data_from_string
tensor2tensor/visualization/visualization_test.py::VisualizationTest::test_session

==================================================================================================== ERRORS =====================================================================================================
_________________________________________________________________ ERROR collecting tensor2tensor/data_generators/ops/pack_sequences_ops_test.py _________________________________________________________________
ImportError while importing test module '/home/saikatd2/projects/borntobeflaky/projects/pytorch-deps/tensor2tensor/tensor2tensor/data_generators/ops/pack_sequences_ops_test.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
tensor2tensor/data_generators/ops/pack_sequences_ops_test.py:22: in <module>
    from tensor2tensor.data_generators.ops import pack_sequences_ops
E   ImportError: cannot import name 'pack_sequences_ops' from 'tensor2tensor.data_generators.ops' (unknown location)
______________________________________________________________ ERROR collecting tensor2tensor/data_generators/ops/subword_text_encoder_ops_test.py ______________________________________________________________
ImportError while importing test module '/home/saikatd2/projects/borntobeflaky/projects/pytorch-deps/tensor2tensor/tensor2tensor/data_generators/ops/subword_text_encoder_ops_test.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
tensor2tensor/data_generators/ops/subword_text_encoder_ops_test.py:22: in <module>
    from tensor2tensor.data_generators.ops import subword_text_encoder_ops
E   ImportError: cannot import name 'subword_text_encoder_ops' from 'tensor2tensor.data_generators.ops' (unknown location)
______________________________________________________________________________ ERROR collecting tensor2tensor/rl/evaluator_test.py ______________________________________________________________________________
tensor2tensor/rl/evaluator_test.py:21: in <module>
    from tensor2tensor.rl import evaluator
tensor2tensor/rl/evaluator.py:50: in <module>
    flags.DEFINE_string("output_dir", "", "Main directory for multi-runs.")
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_core/python/platform/flags.py:58: in wrapper
    return original_function(*args, **kwargs)
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/absl/flags/_defines.py:241: in DEFINE_string
    DEFINE(parser, name, default, help, flag_values, serializer, **args)
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/absl/flags/_defines.py:82: in DEFINE
    flag_values, module_name)
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/absl/flags/_defines.py:104: in DEFINE_flag
    fv[flag.name] = flag
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/absl/flags/_flagvalues.py:430: in __setitem__
    raise _exceptions.DuplicateFlagError.from_flag(name, self)
E   absl.flags._exceptions.DuplicateFlagError: The flag 'output_dir' is defined twice. First from tensor2tensor.bin.t2t_trainer, Second from tensor2tensor.rl.evaluator.  Description from first occurrence: Base output directory for run.
=============================================================================================== warnings summary ================================================================================================
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_core/python/pywrap_tensorflow_internal.py:15
  /home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_core/python/pywrap_tensorflow_internal.py:15: DeprecationWarning: the imp module is deprecated in favour of importlib; see the module's documentation for alternative uses
    import imp

/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/pandas/core/dtypes/inference.py:6
  /home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/pandas/core/dtypes/inference.py:6: DeprecationWarning: Using or importing the ABCs from 'collections' instead of from 'collections.abc' is deprecated since Python 3.3,and in 3.9 it will stop working
    from collections import Iterable

/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/pandas/core/tools/datetimes.py:3
  /home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/pandas/core/tools/datetimes.py:3: DeprecationWarning: Using or importing the ABCs from 'collections' instead of from 'collections.abc' is deprecated since Python 3.3,and in 3.9 it will stop working
    from collections import MutableMapping

/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_core/contrib/learn/python/learn/learn_io/generator_io.py:26
  /home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_core/contrib/learn/python/learn/learn_io/generator_io.py:26: DeprecationWarning: Using or importing the ABCs from 'collections' instead of from 'collections.abc' is deprecated since Python 3.3,and in 3.9 it will stop working
    from collections import Container

/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108
  /home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/tensorflow_gan/python/losses/tuple_losses.py:108: DeprecationWarning: inspect.getargspec() is deprecated since Python 3.0, use inspect.signature() or inspect.getfullargspec()
    argspec = inspect.getargspec(loss_fn)

tensor2tensor/data_generators/gym_env_test.py:37
  /home/saikatd2/projects/borntobeflaky/projects/pytorch-deps/tensor2tensor/tensor2tensor/data_generators/gym_env_test.py:37: PytestCollectionWarning: cannot collect test class 'TestEnv' because it has a __init__ constructor (from: tensor2tensor/data_generators/gym_env_test.py)
    class TestEnv(gym.Env):

tensor2tensor/data_generators/text_problems_test.py:31
  /home/saikatd2/projects/borntobeflaky/projects/pytorch-deps/tensor2tensor/tensor2tensor/data_generators/text_problems_test.py:31: PytestCollectionWarning: cannot collect test class 'Test1' because it has a __init__ constructor (from: tensor2tensor/data_generators/text_problems_test.py)
    class Test1(text_problems.Text2textTmpdir):

/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/kfac/python/keras/optimizers.py:36
  /home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/kfac/python/keras/optimizers.py:36: DeprecationWarning: inspect.getargspec() is deprecated since Python 3.0, use inspect.signature() or inspect.getfullargspec()
    _KFAC_ARGS = inspect.getargspec(optimizer.KfacOptimizer.__init__)

/home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/kfac/python/keras/optimizers.py:37
  /home/saikatd2/anaconda3/envs/pv3.6_tensor2tensor/lib/python3.7/site-packages/kfac/python/keras/optimizers.py:37: DeprecationWarning: inspect.getargspec() is deprecated since Python 3.0, use inspect.signature() or inspect.getfullargspec()
    _PERIODIC_KFAC_ARGS = inspect.getargspec(_KFAC_OPT_CLASS.__init__)

tensor2tensor/utils/data_reader_test.py:37
  /home/saikatd2/projects/borntobeflaky/projects/pytorch-deps/tensor2tensor/tensor2tensor/utils/data_reader_test.py:37: PytestCollectionWarning: cannot collect test class 'TestProblem' because it has a __init__ constructor (from: tensor2tensor/utils/data_reader_test.py)
    class TestProblem(problem_mod.Problem):

-- Docs: https://docs.pytest.org/en/latest/warnings.html
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! Interrupted: 3 errors during collection !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
23 warnings, 3 errors in 7.80s
