experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add an XLA test for ComputationBuilder::SqrtF32(0)

Browse Source 
Change: 144124103
This commit is contained in:
David Majnemer 2017-01-10 13:53:29 -08:00 committed by TensorFlower Gardener
parent a762b0a0de
commit b76344d0ab
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tensorflow/compiler/xla/tests/scalar_computations_test.cc
View File

@ -608,6 +608,20 @@ TEST_F(ScalarComputationsTest, ComplicatedArithmeticExpressionS32) {
ComputeAndCompareR0<int32>(&b, 10, {});
}
TEST_F(ScalarComputationsTest, SqrtF320) {
ComputationBuilder builder(client_, TestName());
Literal zero_literal = LiteralUtil::Zero(PrimitiveType::F32);
std::unique_ptr<GlobalData> zero_data =
client_->TransferToServer(zero_literal).ConsumeValueOrDie();
ComputationDataHandle zero =
builder.Parameter(0, zero_literal.shape(), "zero");
builder.SqrtF32(zero);
ComputeAndCompareR0<float>(&builder, 0.0f, {zero_data.get()}, error_spec_);
}
} // namespace
} // namespace xla