Home
last modified time | relevance | path

Searched refs:a_literal (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_precision_test.cc232 Literal a_literal = LiteralUtil::CreateR1<float>({input_values}); in XLA_TEST_P() local
234 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_P()
235 auto a = Parameter(&builder, 0, a_literal.shape(), "a"); in XLA_TEST_P()
255 Literal a_literal = LiteralUtil::CreateR1<float>({1.00001}); in XLA_TEST_F() local
257 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
258 auto a = Parameter(&builder, 0, a_literal.shape(), "a"); in XLA_TEST_F()
284 Literal a_literal = LiteralUtil::CreateR1<float>({1.00001, 1.00001}); in XLA_TEST_F() local
286 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
287 auto a = Parameter(&builder, 0, a_literal.shape(), "a"); in XLA_TEST_F()
310 Literal a_literal = LiteralUtil::CreateR1<float>({1.00001, 1.00001}); in XLA_TEST_F() local
[all …]
Dscalar_computations_test.cc164 Literal a_literal = LiteralUtil::CreateR0<int64>(value); in XLA_TEST_F() local
166 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
228 Literal a_literal = LiteralUtil::CreateR0<float>(2.1f); in XLA_TEST_F() local
233 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
239 XlaOp a = Parameter(&builder, 0, a_literal.shape(), "a"); in XLA_TEST_F()
Dconcat_test.cc497 auto a_literal = LiteralUtil::CreateR1<float>({256.0}); in XLA_TEST_F() local
498 auto a = Parameter(&builder, 0, a_literal.shape(), "x"); in XLA_TEST_F()
517 auto a_data = client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
Darray_elementwise_ops_test.cc322 Literal a_literal = LiteralUtil::CreateR1<float>({a_values}); in TEST_P() local
324 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in TEST_P()
326 auto a_param = Parameter(&builder, 0, a_literal.shape(), "a_param"); in TEST_P()
331 auto b_param = Parameter(&builder, 1, a_literal.shape(), "b_param"); in TEST_P()
355 auto a_literal = LiteralUtil::CreateR1<float>(values); in XLA_TEST_F() local
356 auto a = Parameter(&builder, 0, a_literal.shape(), "x"); in XLA_TEST_F()
383 auto a_data = client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
2896 Literal a_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_F() local
2898 auto a = ConstantLiteral(&builder, a_literal); in XLA_TEST_F()