Searched defs:OneHotContext (Results 1 – 1 of 1) sorted by relevance
37 struct OneHotContext { struct38 OneHotContext(TfLiteContext* context, TfLiteNode* node) { in OneHotContext() function53 const TfLiteTensor* indices;54 const TfLiteTensor* depth;55 const TfLiteTensor* on_value;56 const TfLiteTensor* off_value;57 TfLiteTensor* output;58 int axis;59 int output_dims;60 TfLiteType dtype;