Searched refs:test_to_device_and_back (Results 1 – 1 of 1) sorted by relevance
835 auto test_to_device_and_back = [this](const Literal& literal) { in XLA_TEST_F() local847 test_to_device_and_back(LiteralUtil::CreateR0<float>(42.0)); in XLA_TEST_F()848 test_to_device_and_back(LiteralUtil::CreateR0<bool>(true)); in XLA_TEST_F()849 test_to_device_and_back(LiteralUtil::CreateR1<float>({1.0, 42.0, 744.4})); in XLA_TEST_F()850 test_to_device_and_back( in XLA_TEST_F()852 test_to_device_and_back(LiteralUtil::CreateR2<int32>({{2, 1}, {4444, 56}})); in XLA_TEST_F()855 test_to_device_and_back(LiteralUtil::MakeTuple({})); in XLA_TEST_F()858 test_to_device_and_back(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()860 test_to_device_and_back(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()865 test_to_device_and_back(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()[all …]