Searched refs:y_literal (Results 1 – 4 of 4) sorted by relevance
79 auto y_literal = xla::LiteralUtil::CreateR0<int32>(32); in TEST() local81 auto y_global_or = client->TransferToServer(y_literal); in TEST()117 auto y_literal = xla::LiteralUtil::CreateR0<int32>(32); in TEST() local119 auto y_global_or = client->TransferToServer(y_literal); in TEST()120 auto unused_global_or = client->TransferToServer(y_literal); in TEST()
810 auto y_literal = LiteralUtil::CreateR0<float>(3.f); in XLA_TEST_F() local812 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()832 auto y_literal = LiteralUtil::CreateR0<float>(1.5f); in XLA_TEST_F() local835 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()860 auto y_literal = LiteralUtil::CreateR0<float>(1.5f); in XLA_TEST_F() local863 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()
307 auto y_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F() local312 auto y_buffer = LiteralToShapedBuffer(y_literal); in XLA_TEST_F()
3065 auto y_literal = LiteralUtil::CreateR1<float>({4, 5}); in XLA_TEST_F() local3067 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()3070 auto y = Parameter(&builder, 1, y_literal.shape(), "y"); in XLA_TEST_F()