Searched defs:upper_diag_index (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 57 int64 upper_diag_index = 0; in ProcessDiagIndex() local
94 const int64 upper_diag_index, in ValidateDiagIndexWithOutputMatrixSize()
120 const int64 upper_diag_index, const int64 max_diag_len, in SetMatrixDiag()
257 int64 upper_diag_index = 0; in Compile() local
370 int64 upper_diag_index = 0; in Compile() local
485 int64 upper_diag_index = 0; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| 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()
|
| 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()
|
| D | matrix_diag_op.cc | 64 int32 upper_diag_index = 0; in Compute() local
168 int32 upper_diag_index = 0; in Compute() local
|
| D | matrix_set_diag_op.cc | 62 int32 upper_diag_index = 0; in Compute() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1246 int32* lower_diag_index, int32* upper_diag_index) { in ReadDiagIndex()
1281 int32 upper_diag_index = 0; in MatrixDiagPartV2Shape() local
1338 int32 upper_diag_index = 0; in MatrixDiagV2Shape() local
1429 int32 upper_diag_index = 0; in MatrixSetDiagV2Shape() local
|