Searched refs:OneHotOpTest (Results 1 – 1 of 1) sorted by relevance
65 TEST(OneHotOpTest, BasicFloat) { in TEST() argument76 TEST(OneHotOpTest, BasicInt) { in TEST() argument86 TEST(OneHotOpTest, BasicBool) { in TEST() argument98 TEST(OneHotOpTest, SmallDepth) { in TEST() argument108 TEST(OneHotOpTest, BigDepth) { in TEST() argument118 TEST(OneHotOpTest, OnOffValues) { in TEST() argument132 TEST(OneHotOpTest, ZeroAxis) { in TEST() argument146 TEST(OneHotOpTest, MultiDimensionalIndices) { in TEST() argument160 TEST(OneHotOpTest, Int64Indices) { in TEST() argument