Searched refs:scale_literal (Results 1 – 1 of 1) sorted by relevance
516 auto scale_literal = LiteralUtil::CreateR1<float>(scale); in XLA_TEST_P() local522 Parameter(&builder, 1, scale_literal.shape(), "offset"); in XLA_TEST_P()533 client_->TransferToServer(scale_literal).ConsumeValueOrDie(); in XLA_TEST_P()614 auto scale_literal = LiteralUtil::CreateR1<float>(scale); in XLA_TEST_P() local622 Parameter(&builder, 1, scale_literal.shape(), "offset"); in XLA_TEST_P()634 client_->TransferToServer(scale_literal).ConsumeValueOrDie(); in XLA_TEST_P()803 auto scale_literal = LiteralUtil::CreateR1<float>(scale); in XLA_TEST_P() local810 auto scale_parameter = Parameter(&builder, 1, scale_literal.shape(), "scale"); in XLA_TEST_P()819 client_->TransferToServer(scale_literal).ConsumeValueOrDie(); in XLA_TEST_P()