Home
last modified time | relevance | path

Searched refs:ShapedBufferToLiteral (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_execute_test.cc62 LiteralTestUtil::ExpectR0Near<float>(123.f, ShapedBufferToLiteral(result), in XLA_TEST_F()
75 LiteralTestUtil::ExpectR0Near<float>(165.f, ShapedBufferToLiteral(result), in XLA_TEST_F()
88 LiteralTestUtil::ExpectR1Near<float>({}, ShapedBufferToLiteral(result), in XLA_TEST_F()
103 {2.0f, 4.0f, 6.0f}, ShapedBufferToLiteral(result), error_spec_); in XLA_TEST_F()
120 {2.0f, 4.0f, 6.0f}, ShapedBufferToLiteral(result), error_spec_); in XLA_TEST_F()
146 ShapedBufferToLiteral(result_colmaj), in XLA_TEST_F()
153 ShapedBufferToLiteral(result_param_swap), in XLA_TEST_F()
179 ShapedBufferToLiteral(result_colmaj), in XLA_TEST_F()
192 ShapedBufferToLiteral(result_rowmaj), in XLA_TEST_F()
214 Literal result_literal = ShapedBufferToLiteral(result); in XLA_TEST_F()
[all …]
Dlocal_client_allocation_test.cc61 {2.0f, 4.0f, 6.0f}, ShapedBufferToLiteral(*result), error_spec_); in XLA_TEST_F()
95 {2.0f, 4.0f, 6.0f}, ShapedBufferToLiteral(result), error_spec_); in XLA_TEST_F()
Dlocal_client_test_base.cc145 Literal LocalClientTestBase::ShapedBufferToLiteral( in ShapedBufferToLiteral() function in xla::LocalClientTestBase
147 return local_client_->ShapedBufferToLiteral(shaped_buffer) in ShapedBufferToLiteral()
Dlocal_client_test_base.h94 Literal ShapedBufferToLiteral(const ShapedBuffer& shaped_buffer);
Dexhaustive_op_test_utils.h277 client_->ShapedBufferToLiteral(result));
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.h162 StatusOr<Literal> ShapedBufferToLiteral(const ShapedBuffer& shaped_buffer);
Dlocal_client.cc415 StatusOr<Literal> LocalClient::ShapedBufferToLiteral( in ShapedBufferToLiteral() function in xla::LocalClient
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc346 client->ShapedBufferToLiteral(*final_result)); in ReplayComputation()