Home
last modified time | relevance | path

Searched refs:NPyCustomFloat_ArgMaxFunc (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/lib/core/
H A Dbfloat16.cc621 int NPyCustomFloat_ArgMaxFunc(void* data, npy_intp n, npy_intp* max_ind, in NPyCustomFloat_ArgMaxFunc() function
1642 arr_funcs.argmax = NPyCustomFloat_ArgMaxFunc<T>; in RegisterNumpyDtype()