Searched refs:kJax (Results 1 – 4 of 4) sorted by relevance
129 EXPECT_EQ(tf_op.category, Category::kJax); in TEST()155 EXPECT_EQ(tf_op.category, Category::kJax); in TEST()
37 kJax, enumerator
91 tf_op = {Category::kJax, parts[0], parts[1]}; in ParseTfOpFullname()
76 if (category == Category::kTensorFlow || category == Category::kJax) { in ProcessTfOpEvent()