OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 12:20:24 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[tf-signal] Fix pip tests by including test_util in signal_py

Browse Source 
PiperOrigin-RevId: 171042732
This commit is contained in:
Shanqing Cai 2017-10-04 12:03:39 -07:00 committed by TensorFlower Gardener
parent f8550f4e94
commit c9915d1a20
1 changed files with 1 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tensorflow/compiler/xla/tests/local_client_execute_test.cc
View File

@ -814,7 +814,7 @@ XLA_TEST_F(LocalClientExecuteTest, ShapeBufferToLiteralConversion) {
test_to_device_and_back(*Literal::CreateR0<bool>(true));
test_to_device_and_back(*Literal::CreateR1<float>({1.0, 42.0, 744.4}));
test_to_device_and_back(
*Literal::CreateR2<float>({{1.0, 2.0, 3.0}, {44.0, 0.1, -3}}));
*Literal::CreateR2<double>({{1.0, 2.0, 3.0}, {44.0, 0.1, -3}}));
test_to_device_and_back(*Literal::CreateR2<int32>({{2, 1}, {4444, 56}}));
// Null shape (empty tuple).
@ -835,30 +835,6 @@ XLA_TEST_F(LocalClientExecuteTest, ShapeBufferToLiteralConversion) {
Literal::CreateR0<bool>(false).get()}));
}
XLA_TEST_F(LocalClientExecuteTest, ShapeBufferToLiteralConversion64bit) {
// Test copying Literals to the device as ShapedBuffers, then copying them
// back again to Literals for 64-bit values.
auto test_to_device_and_back = [this](const Literal& literal) {
TF_ASSERT_OK_AND_ASSIGN(
auto shaped_buffer,
local_client_->LiteralToShapedBuffer(
literal, local_client_->default_device_ordinal(), allocator_));
TF_ASSERT_OK_AND_ASSIGN(
auto transferred_literal,
local_client_->ShapedBufferToLiteral(*shaped_buffer));
EXPECT_EQ(literal, *transferred_literal);
};
test_to_device_and_back(
*Literal::CreateR2<double>({{1.0, 2.0, 3.0}, {44.0, 0.1, -3}}));
test_to_device_and_back(*Literal::CreateR2<int64>({{2, 1}, {4444, 56}}));
test_to_device_and_back(
*Literal::CreateR2<uint64>({{20000000000ULL, 1}, {4444, 56}}));
test_to_device_and_back(
*Literal::MakeTuple({Literal::CreateR1<double>({1.0, -42.0}).get(),
Literal::CreateR0<int64>(123456789000LL).get()}));
}
// Benchmark that measures the overhead of the LocalClient API when running a
// trivial computation
void BM_LocalClientOverhead(int num_iters) {