Searched defs:num_diags (Results 1 – 10 of 10) sorted by relevance
| /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()
85 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
104 const int num_diags, const int max_diag_len, const int lower_diag_index, in MatrixDiagPartKernel()
146 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
|
| D | matrix_set_diag_op_gpu.cu.cc | 47 const int num_threads, const int m, const int n, const int num_diags, in MatrixSetDiagKernel()
73 const int num_threads, const int m, const int n, const int num_diags, in MatrixCopyInputAndSetDiagKernel()
113 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
|
| D | matrix_diag_op.cc | 126 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
203 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
351 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
397 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
|
| D | matrix_set_diag_op.cc | 125 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
216 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
|
| D | tridiagonal_solve_op.cc | 56 auto num_diags = input_matrix_shapes[0].dim_size(0); in ValidateInputMatrixShapes() local
|
| D | tridiagonal_solve_op_gpu.cu.cc | 113 auto num_diags = input_matrix_shapes[0].dim_size(0); in ValidateInputMatrixShapes() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 119 const int64 num_diags, const int64 lower_diag_index, in SetMatrixDiag()
275 const int64 num_diags = upper_diag_index - lower_diag_index + 1; in Compile() local
390 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compile() local
495 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compile() local
|
| /external/llvm-project/lldb/source/Plugins/ExpressionParser/Clang/ |
| D | ClangExpressionParser.cpp | 1262 size_t num_diags = diagnostics.size(); in RewriteExpression() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1350 const int32 num_diags = c->Value(c->Dim(input_shape, input_rank - 2)); in MatrixDiagV2Shape() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1515 int num_diags = k[1] - k[0] + 1; in matchAndRewrite() local
|