Searched defs:OneHot (Results 1 – 6 of 6) sorted by relevance
32 OneHot() : PrimitiveC(prim::kPrimOneHot->name()) { in OneHot() function
41 void OneHot(const S *indices, size_t depth, const T *on_value, const T *off_value, size_t left_dim_… in OneHot() function
57 class OneHot(TensorOperation): class
157 OneHot::OneHot(int32_t num_classes) : data_(std::make_shared<Data>(num_classes)) {} in OneHot() function in mindspore::dataset::transforms::OneHot
535 class OneHot(Cell): class
3568 class OneHot(Primitive): class