Searched defs:lower_diag_index (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 56 int64 lower_diag_index = 0; in ProcessDiagIndex() local
93 const int64 lower_diag_index, in ValidateDiagIndexWithOutputMatrixSize()
119 const int64 num_diags, const int64 lower_diag_index, in SetMatrixDiag()
256 int64 lower_diag_index = 0; in Compile() local
369 int64 lower_diag_index = 0; in Compile() local
484 int64 lower_diag_index = 0; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op_gpu.cu.cc | 47 const int num_diags, const int max_diag_len, const int lower_diag_index, in MatrixDiagKernel()
104 const int num_diags, const int max_diag_len, const int lower_diag_index, in MatrixDiagPartKernel()
|
| D | matrix_diag_op.cc | 63 int32 lower_diag_index = 0; in Compute() local
167 int32 lower_diag_index = 0; in Compute() local
|
| D | matrix_set_diag_op.cc | 61 int32 lower_diag_index = 0; in Compute() local
|
| D | matrix_set_diag_op_gpu.cu.cc | 74 const int max_diag_len, const int lower_diag_index, in MatrixCopyInputAndSetDiagKernel()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1246 int32* lower_diag_index, int32* upper_diag_index) { in ReadDiagIndex()
1280 int32 lower_diag_index = 0; in MatrixDiagPartV2Shape() local
1337 int32 lower_diag_index = 0; in MatrixDiagV2Shape() local
1428 int32 lower_diag_index = 0; in MatrixSetDiagV2Shape() local
|