Searched refs:OneHotCompute (Results 1 – 1 of 1) sorted by relevance
98 void OneHotCompute(const OneHotContext& op_context) { in OneHotCompute() function172 OneHotCompute<float>(op_context); in Eval()175 OneHotCompute<int>(op_context); in Eval()178 OneHotCompute<int64_t>(op_context); in Eval()181 OneHotCompute<int8_t>(op_context); in Eval()184 OneHotCompute<uint8_t>(op_context); in Eval()187 OneHotCompute<bool>(op_context); in Eval()