/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | map_test.cc | 172 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 …]
|
D | params_test.cc | 45 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 …]
|
D | prng_test.cc | 205 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()
|
D | deconstruct_tuple_test.cc | 173 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()
|
D | reshape_test.cc | 120 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()
|
D | array_elementwise_ops_test.cc | 2273 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()
|
D | cpu_gpu_fusion_test.cc | 865 auto param0_literal = in BM_ParallelFusion() local 868 client->LiteralToShapedBuffer(param0_literal, device_ordinal) in BM_ParallelFusion()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | sample_harness.cc | 40 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/ |
D | xla_compiler_test.cc | 221 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/ |
D | hlo_evaluator_test.cc | 3256 Literal param0_literal = LiteralUtil::CreateR1<float>({1, 2, 3, 4}); in TEST_P() local 3261 add, {{param0, ¶m0_literal}, {square, &square_literal}})); in TEST_P()
|