Searched refs:ShapedBufferToLiteral (Results 1 - 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | local_client_allocation_test.cc | 61 {2.0f, 4.0f, 6.0f}, *ShapedBufferToLiteral(*result), error_spec_); 95 {2.0f, 4.0f, 6.0f}, *ShapedBufferToLiteral(*result), error_spec_);
|
H A D | local_client_execute_test.cc | 65 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 D | local_client_test_base.h | 93 std::unique_ptr<Literal> ShapedBufferToLiteral(
|
H A D | local_client_test_base.cc | 138 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 D | local_client.h | 136 StatusOr<std::unique_ptr<Literal>> ShapedBufferToLiteral(
|
H A D | local_client.cc | 284 StatusOr<std::unique_ptr<Literal>> LocalClient::ShapedBufferToLiteral( function in class:xla::LocalClient
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
H A D | local_computation_builder.cc | 125 return client->ShapedBufferToLiteral(*shaped_buffer()).ConsumeValueOrDie(); 214 client->ShapedBufferToLiteral(*result_buffer_status.ValueOrDie());
|
Completed in 267 milliseconds