Searched defs:one_hot (Results 1 – 8 of 8) sorted by relevance
47 def one_hot(nptype): function
36 auto one_hot = std::make_shared<OneHot>(); in TEST_F() local
87 def __init__(self, path, one_hot=True): argument
84 def __init__(self, path, one_hot=True): argument
118 std::shared_ptr<TensorTransform> one_hot = std::make_shared<transforms::OneHot>(10); in TEST_F() local
31 def one_hot(index, depth): function
129 auto one_hot = std::make_shared<transforms::OneHotOperation>(num_classes); in __anon360f34600e02() local
783 OneHot one_hot{}; in ConvertOneHot() local