Searched refs:test_to_device_and_back (Results 1 – 1 of 1) sorted by relevance
784 auto test_to_device_and_back = [this](const Literal& literal) { in XLA_TEST_F() local796 test_to_device_and_back(LiteralUtil::CreateR0<float>(42.0)); in XLA_TEST_F()797 test_to_device_and_back(LiteralUtil::CreateR0<bool>(true)); in XLA_TEST_F()798 test_to_device_and_back(LiteralUtil::CreateR1<float>({1.0, 42.0, 744.4})); in XLA_TEST_F()799 test_to_device_and_back( in XLA_TEST_F()801 test_to_device_and_back(LiteralUtil::CreateR2<int32>({{2, 1}, {4444, 56}})); in XLA_TEST_F()804 test_to_device_and_back(LiteralUtil::MakeTuple({})); in XLA_TEST_F()807 test_to_device_and_back(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()809 test_to_device_and_back(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()814 test_to_device_and_back(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()[all …]