Searched refs:LiteralToShapedBuffer (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 72 auto x_value = LiteralToShapedBuffer(LiteralUtil::CreateR0<float>(42.0f)); in XLA_TEST_F()
85 auto x_array = LiteralToShapedBuffer(LiteralUtil::CreateR1<float>({})); in XLA_TEST_F()
99 LiteralToShapedBuffer(LiteralUtil::CreateR1<float>({0.0f, 1.0f, 2.0f})); in XLA_TEST_F()
113 LiteralToShapedBuffer(LiteralUtil::CreateR1<float>({0.0f, 1.0f, 2.0f})); in XLA_TEST_F()
132 auto x_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
138 auto y_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
164 auto x_array = LiteralToShapedBuffer( in XLA_TEST_F()
166 auto y_array = LiteralToShapedBuffer( in XLA_TEST_F()
203 auto x_array = LiteralToShapedBuffer( in XLA_TEST_F()
205 auto y_array = LiteralToShapedBuffer( in XLA_TEST_F()
[all …]
|
| D | local_client_test_base.cc | 138 ScopedShapedBuffer LocalClientTestBase::LiteralToShapedBuffer( in LiteralToShapedBuffer() function in xla::LocalClientTestBase
141 ->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 | 90 ScopedShapedBuffer LiteralToShapedBuffer(const Literal& literal);
|
| D | cpu_gpu_fusion_test.cc | 868 client->LiteralToShapedBuffer(param0_literal, device_ordinal) in BM_ParallelFusion()
874 client->LiteralToShapedBuffer(param1_literal, device_ordinal) in BM_ParallelFusion()
880 client->LiteralToShapedBuffer(param2_literal, device_ordinal) in BM_ParallelFusion()
|
| D | exhaustive_op_test_utils.h | 259 ->LiteralToShapedBuffer(*input_literal,
|
| D | dot_operation_test.cc | 1845 client->LiteralToShapedBuffer(input_literal, device_ordinal) in DOT_ReorderContracting()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.h | 152 StatusOr<ScopedShapedBuffer> LiteralToShapedBuffer(
|
| D | local_client.cc | 399 StatusOr<ScopedShapedBuffer> LocalClient::LiteralToShapedBuffer( in LiteralToShapedBuffer() function in xla::LocalClient
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tracked_device_buffer_test.cc | 104 client->LiteralToShapedBuffer(literal, /*device_ordinal=*/0)); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 257 client->LiteralToShapedBuffer(literal, /*device_ordinal=*/0)); in ReplayComputation()
|