Searched refs:LiteralToShapedBuffer (Results 1 - 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | local_client_execute_test.cc | 75 auto x_value = LiteralToShapedBuffer(*Literal::CreateR0<float>(42.0f));
89 auto x_array = LiteralToShapedBuffer(*Literal::CreateR1<float>({}));
104 LiteralToShapedBuffer(*Literal::CreateR1<float>({0.0f, 1.0f, 2.0f}));
119 LiteralToShapedBuffer(*Literal::CreateR1<float>({0.0f, 1.0f, 2.0f}));
139 auto x_array = LiteralToShapedBuffer(*Literal::CreateR2WithLayout(
145 auto y_array = LiteralToShapedBuffer(*Literal::CreateR2WithLayout(
171 auto x_array = LiteralToShapedBuffer(
173 auto y_array = LiteralToShapedBuffer(
208 auto x_array = LiteralToShapedBuffer(
210 auto y_array = LiteralToShapedBuffer(
[all...] |
| H A D | local_client_allocation_test.cc | 48 LiteralToShapedBuffer(*Literal::CreateR1<float>({0.0f, 1.0f, 2.0f}));
|
| H A D | local_client_test_base.h | 87 // LocalClient::LiteralToShapedBuffer.
88 std::unique_ptr<ScopedShapedBuffer> LiteralToShapedBuffer(
|
| H A D | local_client_test_base.cc | 131 std::unique_ptr<ScopedShapedBuffer> LocalClientTestBase::LiteralToShapedBuffer( function in class:xla::LocalClientTestBase
134 ->LiteralToShapedBuffer(literal, local_client_->default_device_ordinal())
|
| H A D | fusion_test.cc | 800 client->LiteralToShapedBuffer(*param0_literal, device_ordinal)
806 client->LiteralToShapedBuffer(*param1_literal, device_ordinal)
812 client->LiteralToShapedBuffer(*param2_literal, device_ordinal)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | local_client.h | 130 StatusOr<std::unique_ptr<ScopedShapedBuffer>> LiteralToShapedBuffer(
|
| H A D | local_client.cc | 269 LocalClient::LiteralToShapedBuffer(const Literal& literal, int device_ordinal, function in class:xla::LocalClient
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | local_computation_builder.cc | 103 return client->LiteralToShapedBuffer(arg, device_ordinal,
|
Completed in 562 milliseconds