Home
last modified time | relevance | path

Searched refs:param0_literal (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmap_test.cc172 Literal param0_literal = LiteralUtil::CreateR0<float>(42.0); in TEST_F() local
174 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
176 auto param = Parameter(&builder, 0, param0_literal.shape(), "param0"); in TEST_F()
186 Literal param0_literal = LiteralUtil::CreateR1<float>({}); in XLA_TEST_F() local
188 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
190 auto param = Parameter(&builder, 0, param0_literal.shape(), "param0"); in XLA_TEST_F()
200 Literal param0_literal = in TEST_F() local
203 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
205 auto param = Parameter(&builder, 0, param0_literal.shape(), "param0"); in TEST_F()
214 Literal param0_literal = in TEST_F() local
[all …]
Dparams_test.cc45 Literal param0_literal = LiteralUtil::CreateR0<float>(3.14159f); in XLA_TEST_F() local
47 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
57 Literal param0_literal = LiteralUtil::CreateR1<float>({}); in XLA_TEST_F() local
59 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
69 Literal param0_literal = LiteralUtil::CreateR1<float>({3.14f, -100.25f}); in XLA_TEST_F() local
71 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
82 Literal param0_literal = LiteralUtil::CreateR1U8(str); in XLA_TEST_F() local
84 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
95 Literal param0_literal = in XLA_TEST_F() local
98 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
[all …]
Dprng_test.cc205 Literal param0_literal = in XLA_TEST_F() local
208 client_->TransferToServer(param0_literal)); in XLA_TEST_F()
210 auto param0 = Parameter(&builder, 0, param0_literal.shape(), "param0"); in XLA_TEST_F()
224 ShapeUtil::ElementsIn(param0_literal.shape())); in XLA_TEST_F()
226 EXPECT_GE(actual.data<float>()[i], param0_literal.data<float>()[i]); in XLA_TEST_F()
227 EXPECT_LT(actual.data<float>()[i], param0_literal.data<float>()[i] + 1.0f); in XLA_TEST_F()
Ddeconstruct_tuple_test.cc173 Literal param0_literal = LiteralUtil::CreateR1<float>({3.14f, -100.25f}); in XLA_TEST_F() local
175 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
Dreshape_test.cc120 Literal param0_literal = LiteralUtil::CreateR0<float>(1.0f); in XLA_TEST_P() local
123 auto input, CreateParameterAndTransferLiteral(0, param0_literal, "param0", in XLA_TEST_P()
150 Literal param0_literal = in XLA_TEST_P() local
154 auto input, CreateParameterAndTransferLiteral(0, param0_literal, "param0", in XLA_TEST_P()
Darray_elementwise_ops_test.cc2273 Literal param0_literal = in XLA_TEST_F() local
2276 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
2283 auto p0 = Parameter(&builder, 0, param0_literal.shape(), "param0"); in XLA_TEST_F()
2295 Literal param0_literal = in XLA_TEST_F() local
2298 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
2305 auto p0 = Parameter(&builder, 0, param0_literal.shape(), "param0"); in XLA_TEST_F()
2317 Literal param0_literal = in XLA_TEST_F() local
2320 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
2323 auto p = Parameter(&builder, 0, param0_literal.shape(), "param0"); in XLA_TEST_F()
Dcpu_gpu_fusion_test.cc865 auto param0_literal = in BM_ParallelFusion() local
868 client->LiteralToShapedBuffer(param0_literal, device_ordinal) in BM_ParallelFusion()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsample_harness.cc40 xla::Literal param0_literal = in main() local
43 client->TransferToServer(param0_literal).ConsumeValueOrDie(); in main()
52 auto p0 = Parameter(&builder, 0, param0_literal.shape(), "param0"); in main()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc221 xla::Literal param0_literal = xla::LiteralUtil::CreateR1<int32>({7, 42}); in TEST_F() local
224 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
272 xla::Literal param0_literal = xla::LiteralUtil::CreateR1<int32>({7, 42}); in TEST_F() local
275 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
285 EXPECT_TRUE(xla::LiteralTestUtil::Equal(param0_literal, actual_literal)); in TEST_F()
610 xla::Literal param0_literal = xla::LiteralUtil::CreateR1<int32>({7, 42}); in TEST_F() local
612 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
1025 xla::Literal param0_literal = xla::LiteralUtil::CreateR1<int32>({7, 42}); in RunAndCheckVariablesComputation() local
1028 client->TransferToServer(param0_literal).ConsumeValueOrDie(); in RunAndCheckVariablesComputation()
1305 xla::Literal param0_literal = in TEST_F() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc3256 Literal param0_literal = LiteralUtil::CreateR1<float>({1, 2, 3, 4}); in TEST_P() local
3261 add, {{param0, &param0_literal}, {square, &square_literal}})); in TEST_P()