Searched refs:CompareUFunc (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | bfloat16.cc | 509 void CompareUFunc(char** args, npy_intp* dimensions, npy_intp* steps, in CompareUFunc() function
633 if (!register_ufunc("equal", CompareUFunc<Bfloat16EqFunctor>, in Initialize()
637 if (!register_ufunc("not_equal", CompareUFunc<Bfloat16NeFunctor>, in Initialize()
641 if (!register_ufunc("less", CompareUFunc<Bfloat16LtFunctor>, compare_types)) { in Initialize()
644 if (!register_ufunc("greater", CompareUFunc<Bfloat16GtFunctor>, in Initialize()
648 if (!register_ufunc("less_equal", CompareUFunc<Bfloat16LeFunctor>, in Initialize()
652 if (!register_ufunc("greater_equal", CompareUFunc<Bfloat16GeFunctor>, in Initialize()
|