Searched refs:FillDiagImpl (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | matrix_diag.cc | 64 void FillDiagImpl(const T* in, T* out, const int batch_size, const int row_size, in FillDiagImpl() function
86 FillDiagImpl<T>(GetTensorData<T>(input), GetTensorData<T>(output), batch_size, in FillDiag()
|
| D | matrix_set_diag.cc | 64 void FillDiagImpl(const T* in, const T* diag, T* out, const int batch_size, in FillDiagImpl() function
88 FillDiagImpl<T>(GetTensorData<T>(input), GetTensorData<T>(diag), in FillDiag()
|