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.cc196 Literal param0_literal = in XLA_TEST_F() local
199 client_->TransferToServer(param0_literal)); in XLA_TEST_F()
201 auto param0 = Parameter(&builder, 0, param0_literal.shape(), "param0"); in XLA_TEST_F()
215 ShapeUtil::ElementsIn(param0_literal.shape())); in XLA_TEST_F()
217 EXPECT_GE(actual.data<float>()[i], param0_literal.data<float>()[i]); in XLA_TEST_F()
218 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.cc116 Literal param0_literal = LiteralUtil::CreateR0<float>(1.0f); in XLA_TEST_P() local
118 auto input = CreateParameterAndTransferLiteral(0, param0_literal, "param0", in XLA_TEST_P()
144 Literal param0_literal = in XLA_TEST_P() local
147 auto input = CreateParameterAndTransferLiteral(0, param0_literal, "param0", in XLA_TEST_P()
Darray_elementwise_ops_test.cc2167 Literal param0_literal = in XLA_TEST_F() local
2170 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
2177 auto p0 = Parameter(&builder, 0, param0_literal.shape(), "param0"); in XLA_TEST_F()
2189 Literal param0_literal = in XLA_TEST_F() local
2192 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
2199 auto p0 = Parameter(&builder, 0, param0_literal.shape(), "param0"); in XLA_TEST_F()
2211 Literal param0_literal = in XLA_TEST_F() local
2214 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
2217 auto p = Parameter(&builder, 0, param0_literal.shape(), "param0"); in XLA_TEST_F()
Dfusion_test.cc866 auto param0_literal = in BM_ParallelFusion() local
869 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.cc214 xla::Literal param0_literal = xla::LiteralUtil::CreateR1<int32>({7, 42}); in TEST_F() local
217 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
265 xla::Literal param0_literal = xla::LiteralUtil::CreateR1<int32>({7, 42}); in TEST_F() local
268 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
278 EXPECT_TRUE(xla::LiteralTestUtil::Equal(param0_literal, actual_literal)); in TEST_F()
502 xla::Literal param0_literal = xla::LiteralUtil::CreateR1<int32>({7, 42}); in TEST_F() local
504 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
534 xla::Literal param0_literal = xla::LiteralUtil::CreateR1<int32>({7, 42}); in TEST_F() local
536 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
960 xla::Literal param0_literal = xla::LiteralUtil::CreateR1<int32>({7, 42}); in RunAndCheckVariablesComputation() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc2053 Literal param0_literal = LiteralUtil::CreateR1<float>({1, 2, 3, 4}); in TEST_P() local
2058 add, {{param0, &param0_literal}, {square, &square_literal}})); in TEST_P()