Searched defs:lower_diag_index (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 57 int64_t lower_diag_index = 0; in ProcessDiagIndex() local
94 const int64_t lower_diag_index, in ValidateDiagIndexWithOutputMatrixSize()
121 const int64_t lower_diag_index, in SetMatrixDiag()
259 int64_t lower_diag_index = 0; in Compile() local
372 int64_t lower_diag_index = 0; in Compile() local
487 int64_t 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_t lower_diag_index = 0; in Compute() local
173 int32_t lower_diag_index = 0; in Compute() local
|
| D | matrix_set_diag_op.cc | 61 int32_t 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 | 1337 int32* lower_diag_index, int32* upper_diag_index) { in ReadDiagIndex()
1371 int32_t lower_diag_index = 0; in MatrixDiagPartV2Shape() local
1428 int32_t lower_diag_index = 0; in MatrixDiagV2Shape() local
1519 int32_t lower_diag_index = 0; in MatrixSetDiagV2Shape() local
|