Searched refs:LiteralToShapedBuffer (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 71 auto x_value = LiteralToShapedBuffer(LiteralUtil::CreateR0<float>(42.0f)); in XLA_TEST_F()
84 auto x_array = LiteralToShapedBuffer(LiteralUtil::CreateR1<float>({})); in XLA_TEST_F()
98 LiteralToShapedBuffer(LiteralUtil::CreateR1<float>({0.0f, 1.0f, 2.0f})); in XLA_TEST_F()
112 LiteralToShapedBuffer(LiteralUtil::CreateR1<float>({0.0f, 1.0f, 2.0f})); in XLA_TEST_F()
131 auto x_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
137 auto y_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
163 auto x_array = LiteralToShapedBuffer( in XLA_TEST_F()
165 auto y_array = LiteralToShapedBuffer( in XLA_TEST_F()
200 auto x_array = LiteralToShapedBuffer( in XLA_TEST_F()
202 auto y_array = LiteralToShapedBuffer( in XLA_TEST_F()
[all …]
|
| D | local_client_test_base.cc | 132 ScopedShapedBuffer LocalClientTestBase::LiteralToShapedBuffer( in LiteralToShapedBuffer() function in xla::LocalClientTestBase
135 ->LiteralToShapedBuffer(literal, local_client_->default_device_ordinal()) in LiteralToShapedBuffer()
|
| D | local_client_allocation_test.cc | 48 LiteralToShapedBuffer(LiteralUtil::CreateR1<float>({0.0f, 1.0f, 2.0f})); in XLA_TEST_F()
|
| D | local_client_test_base.h | 85 ScopedShapedBuffer LiteralToShapedBuffer(const Literal& literal);
|
| D | fusion_test.cc | 869 client->LiteralToShapedBuffer(param0_literal, device_ordinal) in BM_ParallelFusion()
875 client->LiteralToShapedBuffer(param1_literal, device_ordinal) in BM_ParallelFusion()
881 client->LiteralToShapedBuffer(param2_literal, device_ordinal) in BM_ParallelFusion()
|
| D | exhaustive_op_test.cc | 301 client_->LiteralToShapedBuffer(input_literal, /*device_ordinal=*/0)); in BuildAndRunComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.h | 128 StatusOr<ScopedShapedBuffer> LiteralToShapedBuffer(
|
| D | local_client.cc | 262 StatusOr<ScopedShapedBuffer> LocalClient::LiteralToShapedBuffer( in LiteralToShapedBuffer() function in xla::LocalClient
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 225 client->LiteralToShapedBuffer(literal, /*device_ordinal=*/0)); in ReplayComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 104 return client.client()->LiteralToShapedBuffer( in FromLiteral()
|