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

Internal tests cleanup.

Browse Source 
PiperOrigin-RevId: 339270788
Change-Id: I381df09a14cd40417797f15e89d419fecdbd1cec
This commit is contained in:
A. Unique TensorFlower 2020-10-27 09:58:31 -07:00 committed by TensorFlower Gardener
parent e8f54a3fd2
commit 0897b9f7cb
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/jit/xla_compilation_cache_test.cc
View File

@ -78,7 +78,9 @@ TEST(XlaCompilationCacheTest, TestDisabledXlaCompilation) {
absl::StrContains(status.error_message(), "XLA compilation disabled"));
}
static void BM_BuildSignature(int iters, int n_args) {
void BM_BuildSignature(::testing::benchmark::State& state) {
const int n_args = state.range(0);
NameAttrList fn;
fn.set_name("afunction");
for (int i = 0; i < n_args; i++) {
@ -93,7 +95,7 @@ static void BM_BuildSignature(int iters, int n_args) {
args[i].constant_value = Tensor(DT_INT32, {4, 0});
}
while (--iters > 0) {
for (auto i : state) {
xla::StatusOr<XlaCompilationCache::Signature> s =
XlaCompilationCache::BuildSignature(fn, args);
CHECK(s.ok());