Home | History | Annotate | Download | only in util

Lines Matching defs:sts

133   SavedTensorSlices sts;
134 EXPECT_TRUE(ParseProtoUnlimited(&sts, value));
135 EXPECT_FALSE(sts.has_meta());
136 *ss = sts.data();
152 SavedTensorSlices sts;
153 EXPECT_TRUE(ParseProtoUnlimited(&sts, value));
155 EXPECT_TRUE(sts.has_meta());
156 EXPECT_EQ(4, sts.meta().tensor_size());
159 EXPECT_EQ(TF_CHECKPOINT_VERSION, sts.meta().versions().producer());
161 sts.meta().versions().min_consumer());
163 EXPECT_FALSE(sts.has_data());
168 const SavedSliceMeta& ssm = sts.meta().tensor(0);
183 const SavedSliceMeta& ssm = sts.meta().tensor(1);
196 const SavedSliceMeta& ssm = sts.meta().tensor(2);
209 const SavedSliceMeta& ssm = sts.meta().tensor(3);