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()
564 auto y_literal = LiteralUtil::CreateR0<float>(3.f); in XLA_TEST_F() local566 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()586 auto y_literal = LiteralUtil::CreateR0<float>(1.5f); in XLA_TEST_F() local589 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()614 auto y_literal = LiteralUtil::CreateR0<float>(1.5f); in XLA_TEST_F() local617 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()
304 auto y_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F() local309 auto y_buffer = LiteralToShapedBuffer(y_literal); in XLA_TEST_F()
2959 auto y_literal = LiteralUtil::CreateR1<float>({4, 5}); in XLA_TEST_F() local2961 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()2964 auto y = Parameter(&builder, 1, y_literal.shape(), "y"); in XLA_TEST_F()