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

Remove possibility to generate a fake token.

Browse Source 
The HloVerifier checks that we never have a token as an entry parameter.
So there should be no need to generate a fake token parameter.

PiperOrigin-RevId: 287966759
Change-Id: Ie45201974d3619e6b2f23f204c8ec9e911473341
This commit is contained in:
Adrian Kuegel 2020-01-03 02:54:18 -08:00 committed by TensorFlower Gardener
parent a0db54f7cb
commit c042ddcbb1
2 changed files with 0 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/xla/tests/test_utils.cc
View File

@ -341,9 +341,6 @@ StatusOr<Literal> MakeFakeLiteralInternal(const Shape& shape,
}));
break;
}
// Token requires no data.
case TOKEN:
break;
default:
return Unimplemented("Unsupported type for fake literal generation: %s",
ShapeUtil::HumanString(shape));

18
tensorflow/compiler/xla/tests/test_utils_test.cc
View File

@ -56,24 +56,6 @@ XLA_TEST_F(TestUtilsTest, UnusedParam) {
TF_ASSERT_OK(MakeFakeArguments(&module).status());
}
XLA_TEST_F(TestUtilsTest, Token) {
auto module = ParseAndReturnUnverifiedModule(
R"(HloModule outfeed_module
ENTRY InfeedToOutfeed {
token0 = token[] parameter(0)
infeed = ((u32[3]{0}, pred[]), token[]) infeed(token0)
infeed.data = (u32[3]{0}, pred[]) get-tuple-element(infeed), index=0
outfeed = token[] outfeed(infeed.data, token0)
ROOT infeed.1 = ((u32[3]{0}, pred[]), token[]) infeed(token0)
infeed.1.data = (u32[3]{0}, pred[]) get-tuple-element(infeed.1), index=0
infeed.1.token = token[] get-tuple-element(infeed.1), index=1
outfeed.1 = token[] outfeed(infeed.1.data, infeed.1.token)
})")
.ValueOrDie();
TF_ASSERT_OK(MakeFakeArguments(module.get()).status());
}
XLA_TEST_F(TestUtilsTest, MultipleIndexSpacesForDynamicSlices) {
auto module = ParseAndReturnVerifiedModule(
R"(HloModule index_space_module