Searched refs:ShapedBufferToLiteral (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dlocal_client_allocation_test.cc61 {2.0f, 4.0f, 6.0f}, *ShapedBufferToLiteral(*result), error_spec_);
95 {2.0f, 4.0f, 6.0f}, *ShapedBufferToLiteral(*result), error_spec_);
H A Dlocal_client_execute_test.cc65 LiteralTestUtil::ExpectR0Near<float>(123.f, *ShapedBufferToLiteral(*result),
79 LiteralTestUtil::ExpectR0Near<float>(165.f, *ShapedBufferToLiteral(*result),
93 LiteralTestUtil::ExpectR1Near<float>({}, *ShapedBufferToLiteral(*result),
109 {2.0f, 4.0f, 6.0f}, *ShapedBufferToLiteral(*result), error_spec_);
127 {2.0f, 4.0f, 6.0f}, *ShapedBufferToLiteral(*result), error_spec_);
153 *ShapedBufferToLiteral(*result_colmaj),
161 *ShapedBufferToLiteral(*result_param_swap), error_spec_);
185 *ShapedBufferToLiteral(*result_colmaj),
197 *ShapedBufferToLiteral(*result_rowmaj),
219 std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*resul
[all...]
H A Dlocal_client_test_base.h93 std::unique_ptr<Literal> ShapedBufferToLiteral(
H A Dlocal_client_test_base.cc138 std::unique_ptr<Literal> LocalClientTestBase::ShapedBufferToLiteral( function in class:xla::LocalClientTestBase
140 return local_client_->ShapedBufferToLiteral(shaped_buffer)
/external/tensorflow/tensorflow/compiler/xla/client/
H A Dlocal_client.h136 StatusOr<std::unique_ptr<Literal>> ShapedBufferToLiteral(
H A Dlocal_client.cc284 StatusOr<std::unique_ptr<Literal>> LocalClient::ShapedBufferToLiteral( function in class:xla::LocalClient
/external/tensorflow/tensorflow/compiler/xla/python/
H A Dlocal_computation_builder.cc125 return client->ShapedBufferToLiteral(*shaped_buffer()).ConsumeValueOrDie();
214 client->ShapedBufferToLiteral(*result_buffer_status.ValueOrDie());

Completed in 267 milliseconds