Searched refs:test_to_device_and_back (Results 1 – 1 of 1) sorted by relevance
820 auto test_to_device_and_back = [this](const Literal& literal) { in XLA_TEST_F() local832 test_to_device_and_back(LiteralUtil::CreateR0<float>(42.0)); in XLA_TEST_F()833 test_to_device_and_back(LiteralUtil::CreateR0<bool>(true)); in XLA_TEST_F()834 test_to_device_and_back(LiteralUtil::CreateR1<float>({1.0, 42.0, 744.4})); in XLA_TEST_F()835 test_to_device_and_back( in XLA_TEST_F()837 test_to_device_and_back(LiteralUtil::CreateR2<int32_t>({{2, 1}, {4444, 56}})); in XLA_TEST_F()840 test_to_device_and_back(LiteralUtil::MakeTuple({})); in XLA_TEST_F()843 test_to_device_and_back(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()845 test_to_device_and_back(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()850 test_to_device_and_back(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()[all …]