Searched refs:b_literal (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | scalar_computations_test.cc | 229 Literal b_literal = LiteralUtil::CreateR0<float>(5.5f); in XLA_TEST_F() local
235 client_->TransferToServer(b_literal).ConsumeValueOrDie(); in XLA_TEST_F()
240 XlaOp b = Parameter(&builder, 1, b_literal.shape(), "b"); in XLA_TEST_F()
|
| D | reduce_test.cc | 869 Literal b_literal = LiteralUtil::CreateR1<float>({1.0f, 4.0f}); in XLA_TEST_F() local
871 client_->TransferToServer(b_literal).ConsumeValueOrDie(); in XLA_TEST_F()
872 auto b = Parameter(&builder, 0, b_literal.shape(), "b"); in XLA_TEST_F()
|
| D | array_elementwise_ops_test.cc | 328 Literal b_literal = LiteralUtil::CreateR1<float>({b_values}); in TEST_P() local
330 client_->TransferToServer(b_literal).ConsumeValueOrDie(); in TEST_P()
357 auto b_literal = LiteralUtil::CreateR1<float>(values); in XLA_TEST_F() local
358 auto b = Parameter(&builder, 1, b_literal.shape(), "x"); in XLA_TEST_F()
384 auto b_data = client_->TransferToServer(b_literal).ConsumeValueOrDie(); in XLA_TEST_F()
|