Home
last modified time | relevance | path

Searched refs:param1_literal (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmap_test.cc349 Literal param1_literal = in TEST_F() local
352 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in TEST_F()
355 auto param1 = Parameter(&builder, 1, param1_literal.shape(), "param1"); in TEST_F()
373 Literal param1_literal = LiteralUtil::CreateR2WithLayout( in XLA_TEST_F() local
376 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in XLA_TEST_F()
379 auto param1 = Parameter(&builder, 1, param1_literal.shape(), "param1"); in XLA_TEST_F()
399 Literal param1_literal = in XLA_TEST_F() local
402 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in XLA_TEST_F()
405 auto param1 = Parameter(&builder, 1, param1_literal.shape(), "param1"); in XLA_TEST_F()
420 Literal param1_literal = in TEST_F() local
[all …]
Darray_elementwise_ops_test.cc2172 Literal param1_literal = in XLA_TEST_F() local
2175 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in XLA_TEST_F()
2178 auto p1 = Parameter(&builder, 1, param1_literal.shape(), "param1"); in XLA_TEST_F()
2194 Literal param1_literal = in XLA_TEST_F() local
2197 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in XLA_TEST_F()
2200 auto p1 = Parameter(&builder, 1, param1_literal.shape(), "param1"); in XLA_TEST_F()
Dfusion_test.cc872 auto param1_literal = in BM_ParallelFusion() local
875 client->LiteralToShapedBuffer(param1_literal, device_ordinal) in BM_ParallelFusion()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsample_harness.cc45 xla::Literal param1_literal = xla::LiteralUtil::CreateR2<float>( in main() local
48 client->TransferToServer(param1_literal).ConsumeValueOrDie(); in main()
53 auto p1 = Parameter(&builder, 1, param1_literal.shape(), "param1"); in main()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc215 xla::Literal param1_literal = xla::LiteralUtil::CreateR1<int32>({-3, 101}); in TEST_F() local
219 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in TEST_F()
266 xla::Literal param1_literal = xla::LiteralUtil::CreateR1<int32>({-3, 101}); in TEST_F() local
270 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in TEST_F()
961 xla::Literal param1_literal = xla::LiteralUtil::CreateR1<int32>({-3, 101}); in RunAndCheckVariablesComputation() local
965 client->TransferToServer(param1_literal).ConsumeValueOrDie(); in RunAndCheckVariablesComputation()
1120 xla::Literal param1_literal = xla::LiteralUtil::CreateR1<int32>({-3, 101}); in TEST_F() local
1122 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in TEST_F()
1239 xla::Literal param1_literal = in TEST_F() local
1244 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in TEST_F()
[all …]