Searched defs:upper_diag_index (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | matrix_diag_ops.cc | 58 int64_t upper_diag_index = 0; in ProcessDiagIndex() local
95 const int64_t upper_diag_index, in ValidateDiagIndexWithOutputMatrixSize()
122 const int64_t upper_diag_index, in SetMatrixDiag()
260 int64_t upper_diag_index = 0; in Compile() local
373 int64_t upper_diag_index = 0; in Compile() local
488 int64_t upper_diag_index = 0; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | matrix_set_diag_op_gpu.cu.cc | 48 const int max_diag_len, const int upper_diag_index, in MatrixSetDiagKernel()
75 const int upper_diag_index, const bool left_align_superdiagonal, in MatrixCopyInputAndSetDiagKernel()
|
| H A D | matrix_diag_op_gpu.cu.cc | 48 const int upper_diag_index, const T padding_value, in MatrixDiagKernel()
105 const int upper_diag_index, const T padding_value, in MatrixDiagPartKernel()
|
| H A D | matrix_diag_op.cc | 64 int32_t upper_diag_index = 0; in Compute() local
174 int32_t upper_diag_index = 0; in Compute() local
|
| H A D | matrix_set_diag_op.cc | 62 int32_t upper_diag_index = 0; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 1337 int32* lower_diag_index, int32* upper_diag_index) { in ReadDiagIndex()
1372 int32_t upper_diag_index = 0; in MatrixDiagPartV2Shape() local
1429 int32_t upper_diag_index = 0; in MatrixDiagV2Shape() local
1520 int32_t upper_diag_index = 0; in MatrixSetDiagV2Shape() local
|