Searched defs:is_arg_max (Results 1 – 4 of 4) sorted by relevance
27 std::function<bool(T, T)> GetComparefunction(bool is_arg_max) { in GetComparefunction()80 T2* output_data, const bool is_arg_max) { in ArgMinMax()
127 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node, bool is_arg_max) { in Eval()
4993 const bool is_arg_max) { in ArgMinMax()
7984 T2* output_data, const bool is_arg_max) { in ArgMinMax()