Searched refs:FillDiag (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | matrix_diag.cc | 84 void FillDiag(const TfLiteTensor* input, TfLiteTensor* output, in FillDiag() function
102 return FillDiag<int64_t>(input, output, batch_size, row_size, col_size); in FillDiagHelper()
105 return FillDiag<int32_t>(input, output, batch_size, row_size, col_size); in FillDiagHelper()
108 return FillDiag<int16_t>(input, output, batch_size, row_size, col_size); in FillDiagHelper()
111 return FillDiag<int8_t>(input, output, batch_size, row_size, col_size); in FillDiagHelper()
114 return FillDiag<uint8_t>(input, output, batch_size, row_size, col_size); in FillDiagHelper()
117 return FillDiag<float>(input, output, batch_size, row_size, col_size); in FillDiagHelper()
|
| D | matrix_set_diag.cc | 85 void FillDiag(const TfLiteTensor* input, const TfLiteTensor* diag, in FillDiag() function
106 return FillDiag<int64_t>(input, diag, output, batch_size, row_size, in FillDiagHelper()
110 return FillDiag<int32_t>(input, diag, output, batch_size, row_size, in FillDiagHelper()
114 return FillDiag<int16_t>(input, diag, output, batch_size, row_size, in FillDiagHelper()
118 return FillDiag<int8_t>(input, diag, output, batch_size, row_size, in FillDiagHelper()
122 return FillDiag<uint8_t>(input, diag, output, batch_size, row_size, in FillDiagHelper()
126 return FillDiag<float>(input, diag, output, batch_size, row_size, in FillDiagHelper()
|