Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dmatrix_diag.cc91 void FillDiagHelper(const TfLiteTensor* input, TfLiteTensor* output) { in FillDiagHelper() function
127 FillDiagHelper(input, output); in Eval()
Dmatrix_set_diag.cc94 void FillDiagHelper(const TfLiteTensor* input, const TfLiteTensor* diag, in FillDiagHelper() function
140 FillDiagHelper(input, diag, output); in Eval()